kiyoonkim commited on
Commit
42608e6
1 Parent(s): d5ae64a

Upload config.yaml with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.yaml +144 -0
config.yaml ADDED
@@ -0,0 +1,144 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ aug_model:
2
+ _target_: mlproject.aug_models.augmentation.VideoAugmentation
3
+ callbacks:
4
+ hf_uploader:
5
+ _target_: mlproject.callbacks.UploadCheckpointsToHuggingFace
6
+ repo_name: aal
7
+ repo_owner: kiyoonkim
8
+ code_dir: /home/kiyoon/project/mlproject/tools
9
+ current_experiment_dir: /disk/scratch2/experiments/aal
10
+ data_dir: /disk/scratch_fast1/datasets
11
+ dataloader:
12
+ _target_: torch.utils.data.dataloader.DataLoader
13
+ batch_sampler: null
14
+ batch_size: 8
15
+ collate_fn: null
16
+ dataset: null
17
+ drop_last: false
18
+ generator: null
19
+ multiprocessing_context: null
20
+ num_workers: 4
21
+ persistent_workers: false
22
+ pin_memory: true
23
+ pin_memory_device: ''
24
+ prefetch_factor: null
25
+ sampler: null
26
+ shuffle: true
27
+ timeout: 0.0
28
+ worker_init_fn: null
29
+ dataset:
30
+ _target_: mlproject.datasets.config.DatasetConfig
31
+ class_keys: null
32
+ num_classes: 51
33
+ download_checkpoint_with_name: null
34
+ download_latest: true
35
+ eval_batch_size: 8
36
+ exp_name: aal
37
+ freeze_backbone: false
38
+ hf_repo_dir: /disk/scratch2/experiments/aal/repo
39
+ hf_username: kiyoonkim
40
+ input_num_frames: 8
41
+ learner:
42
+ _target_: mlproject.boilerplate.Learner
43
+ callbacks: null
44
+ checkpoint_after_validation: true
45
+ checkpoint_every_n_steps: 500
46
+ evaluate_every_n_epochs: null
47
+ evaluate_every_n_steps: 500
48
+ evaluators: null
49
+ experiment_dir: /disk/scratch2/experiments/aal/repo
50
+ experiment_name: aal
51
+ limit_train_iters: null
52
+ limit_val_iters: null
53
+ model: null
54
+ print_model_parameters: false
55
+ resume: false
56
+ test_dataloaders: null
57
+ train_dataloader: null
58
+ train_epochs: null
59
+ train_iters: 10000
60
+ trainers: null
61
+ val_dataloaders: null
62
+ model:
63
+ _target_: mlproject.models.tsm_resnet50.build_tsm_model
64
+ num_classes: 51
65
+ num_frames: 8
66
+ pretrained: kinetics400
67
+ num_workers: 4
68
+ optimizer:
69
+ _target_: torch.optim.sgd.SGD
70
+ dampening: 0
71
+ differentiable: false
72
+ foreach: null
73
+ lr: 1.0e-05
74
+ maximize: false
75
+ momentum: 0.9
76
+ nesterov: false
77
+ params: ???
78
+ weight_decay: 0.0005
79
+ print_config: false
80
+ repo_path: kiyoonkim/aal
81
+ resume: false
82
+ resume_from_checkpoint: null
83
+ root_experiment_dir: /disk/scratch2/experiments
84
+ scheduler:
85
+ _target_: torch.optim.lr_scheduler.MultiStepLR
86
+ gamma: 0.3
87
+ last_epoch: -1
88
+ milestones:
89
+ - 5000
90
+ - 8000
91
+ optimizer: ???
92
+ verbose: false
93
+ seed: 42
94
+ test: false
95
+ torch_dataset:
96
+ _target_: mlproject.torch_datasets.gulp_sparsesample.build_gulp_dataset
97
+ bgr: false
98
+ crop_size: 224
99
+ data_dir: /disk/scratch_fast1/datasets/hmdb51
100
+ dataset_name: hmdb51-gulprgb
101
+ ensure_installed: true
102
+ mean:
103
+ - 0.485
104
+ - 0.456
105
+ - 0.406
106
+ normalise: true
107
+ sets_to_include: null
108
+ std:
109
+ - 0.229
110
+ - 0.224
111
+ - 0.225
112
+ test_num_spatial_crops: 1
113
+ test_scale: 256
114
+ train_horizontal_flip: true
115
+ train_jitter_max: 336
116
+ train_jitter_min: 224
117
+ train: true
118
+ train_batch_size: 8
119
+ wandb_args:
120
+ _target_: wandb.sdk.wandb_init.init
121
+ allow_val_change: null
122
+ anonymous: null
123
+ config: null
124
+ config_exclude_keys: null
125
+ config_include_keys: null
126
+ dir: /disk/scratch2/experiments/aal
127
+ entity: null
128
+ force: null
129
+ group: null
130
+ id: null
131
+ job_type: null
132
+ magic: null
133
+ mode: null
134
+ monitor_gym: null
135
+ name: null
136
+ notes: null
137
+ project: simple-tutorial
138
+ reinit: null
139
+ resume: allow
140
+ save_code: true
141
+ settings: null
142
+ sync_tensorboard: null
143
+ tags: null
144
+ tensorboard: null