kiyoonkim commited on
Commit
66745b9
·
1 Parent(s): 0fd7721

Upload config.yaml with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.yaml +131 -0
config.yaml ADDED
@@ -0,0 +1,131 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ callbacks:
2
+ hf_uploader:
3
+ _target_: mlproject.callbacks.UploadCheckpointsToHuggingFace
4
+ repo_name: aah
5
+ repo_owner: kiyoonkim
6
+ code_dir: /home/kiyoon/project/mlproject
7
+ current_experiment_dir: /disk/scratch2/experiments/aah
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: aah
35
+ freeze_backbone: false
36
+ hf_repo_dir: /disk/scratch2/experiments/aah/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/aah/repo
48
+ experiment_name: aah
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-06
78
+ maximize: false
79
+ params: ???
80
+ weight_decay: 1.0e-05
81
+ print_config: false
82
+ repo_path: kiyoonkim/aah
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.1
89
+ last_epoch: -1
90
+ milestones:
91
+ - 2000
92
+ - 4000
93
+ - 6000
94
+ - 8000
95
+ optimizer: ???
96
+ verbose: false
97
+ seed: 42
98
+ test: false
99
+ torch_dataset:
100
+ _target_: mlproject.torch_datasets.gulp_sparsesample.build_gulp_dataset
101
+ data_dir: /disk/scratch_fast1/datasets
102
+ dataset_name: hmdb51-gulprgb
103
+ sets_to_include: null
104
+ train: true
105
+ train_batch_size: 8
106
+ wandb_args:
107
+ _target_: wandb.sdk.wandb_init.init
108
+ allow_val_change: null
109
+ anonymous: null
110
+ config: null
111
+ config_exclude_keys: null
112
+ config_include_keys: null
113
+ dir: /disk/scratch2/experiments/aah
114
+ entity: null
115
+ force: null
116
+ group: null
117
+ id: null
118
+ job_type: null
119
+ magic: null
120
+ mode: null
121
+ monitor_gym: null
122
+ name: null
123
+ notes: null
124
+ project: simple-tutorial
125
+ reinit: null
126
+ resume: allow
127
+ save_code: true
128
+ settings: null
129
+ sync_tensorboard: null
130
+ tags: null
131
+ tensorboard: null