kiyoonkim commited on
Commit
1c2721f
·
1 Parent(s): 8a9e591

Upload config.yaml with huggingface_hub

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