| callbacks: | |
| hf_uploader: | |
| _target_: mlproject.callbacks.UploadCheckpointsToHuggingFace | |
| repo_name: aae | |
| repo_owner: kiyoonkim | |
| code_dir: /home/kiyoon/project/mlproject | |
| current_experiment_dir: /disk/scratch2/experiments/aae | |
| data_dir: /disk/scratch_fast1/datasets | |
| dataloader: | |
| _target_: torch.utils.data.dataloader.DataLoader | |
| batch_sampler: null | |
| batch_size: 8 | |
| collate_fn: null | |
| dataset: null | |
| drop_last: false | |
| generator: null | |
| multiprocessing_context: null | |
| num_workers: 4 | |
| persistent_workers: false | |
| pin_memory: true | |
| pin_memory_device: '' | |
| prefetch_factor: null | |
| sampler: null | |
| shuffle: true | |
| timeout: 0.0 | |
| worker_init_fn: null | |
| dataset: | |
| _target_: mlproject.data.build_gulp_dataset | |
| data_dir: /disk/scratch_fast1/datasets | |
| dataset_name: hmdb51-gulprgb | |
| sets_to_include: null | |
| download_checkpoint_with_name: null | |
| download_latest: true | |
| eval_batch_size: 8 | |
| exp_name: aae | |
| freeze_backbone: false | |
| hf_repo_dir: /disk/scratch2/experiments/aae/repo | |
| hf_username: kiyoonkim | |
| learner: | |
| _target_: mlproject.boilerplate.Learner | |
| callbacks: null | |
| checkpoint_after_validation: true | |
| checkpoint_every_n_steps: 500 | |
| evaluate_every_n_epochs: null | |
| evaluate_every_n_steps: 500 | |
| evaluators: null | |
| experiment_dir: /disk/scratch2/experiments/aae/repo | |
| experiment_name: aae | |
| limit_train_iters: null | |
| limit_val_iters: null | |
| model: null | |
| print_model_parameters: false | |
| resume: false | |
| test_dataloaders: null | |
| train_dataloader: null | |
| train_epochs: null | |
| train_iters: 10000 | |
| trainers: null | |
| val_dataloaders: null | |
| model: | |
| _target_: mlproject.models.build_tsm_model | |
| model_name: tsm_resnet50 | |
| num_classes: 51 | |
| num_frames: 8 | |
| pretrained: kinetics400 | |
| num_workers: 4 | |
| optimizer: | |
| _partial_: true | |
| _target_: torch.optim.adamw.AdamW | |
| amsgrad: false | |
| betas: | |
| - 0.9 | |
| - 0.999 | |
| capturable: false | |
| differentiable: false | |
| eps: 1.0e-08 | |
| foreach: null | |
| fused: null | |
| lr: 1.0e-05 | |
| maximize: false | |
| weight_decay: 1.0e-05 | |
| print_config: false | |
| repo_path: kiyoonkim/aae | |
| resume: false | |
| resume_from_checkpoint: null | |
| root_experiment_dir: /disk/scratch2/experiments | |
| scheduler: | |
| _partial_: true | |
| _target_: timm.scheduler.cosine_lr.CosineLRScheduler | |
| cycle_decay: 1.0 | |
| cycle_limit: 1 | |
| cycle_mul: 1.0 | |
| initialize: true | |
| k_decay: 1.0 | |
| lr_min: 0.0 | |
| noise_pct: 0.67 | |
| noise_range_t: null | |
| noise_seed: 42 | |
| noise_std: 1.0 | |
| t_in_epochs: true | |
| warmup_lr_init: 0 | |
| warmup_prefix: false | |
| warmup_t: 0 | |
| seed: 42 | |
| test: false | |
| train: true | |
| train_batch_size: 8 | |
| wandb_args: | |
| _target_: wandb.sdk.wandb_init.init | |
| allow_val_change: null | |
| anonymous: null | |
| config: null | |
| config_exclude_keys: null | |
| config_include_keys: null | |
| dir: /disk/scratch2/experiments/aae | |
| entity: null | |
| force: null | |
| group: null | |
| id: null | |
| job_type: null | |
| magic: null | |
| mode: null | |
| monitor_gym: null | |
| name: null | |
| notes: null | |
| project: simple-tutorial | |
| reinit: null | |
| resume: allow | |
| save_code: true | |
| settings: null | |
| sync_tensorboard: null | |
| tags: null | |
| tensorboard: null | |