File size: 3,118 Bytes
a56d54f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
callbacks:
  hf_uploader:
    _target_: mlproject.callbacks.UploadCheckpointsToHuggingFace
    repo_name: aak
    repo_owner: kiyoonkim
code_dir: /home/kiyoon/project/mlproject
current_experiment_dir: /disk/scratch2/experiments/aak
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.datasets.config.DatasetConfig
  class_keys: null
  num_classes: 51
download_checkpoint_with_name: null
download_latest: true
eval_batch_size: 8
exp_name: aak
freeze_backbone: false
hf_repo_dir: /disk/scratch2/experiments/aak/repo
hf_username: kiyoonkim
input_num_frames: 8
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/aak/repo
  experiment_name: aak
  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.tsm_resnet50.build_tsm_model
  num_classes: 51
  num_frames: 8
  pretrained: kinetics400
num_workers: 4
optimizer:
  _target_: torch.optim.sgd.SGD
  dampening: 0
  differentiable: false
  foreach: null
  lr: 1.0e-05
  maximize: false
  momentum: 0.9
  nesterov: false
  params: ???
  weight_decay: 0.0005
print_config: false
repo_path: kiyoonkim/aak
resume: false
resume_from_checkpoint: null
root_experiment_dir: /disk/scratch2/experiments
scheduler:
  _target_: torch.optim.lr_scheduler.MultiStepLR
  gamma: 0.3
  last_epoch: -1
  milestones:
  - 5000
  - 8000
  optimizer: ???
  verbose: false
seed: 42
test: false
torch_dataset:
  _target_: mlproject.torch_datasets.gulp_sparsesample.build_gulp_dataset
  bgr: false
  crop_size: 224
  data_dir: /disk/scratch_fast1/datasets
  dataset_name: hmdb51-gulprgb
  mean:
  - 0.485
  - 0.456
  - 0.406
  normalise: true
  sets_to_include: null
  std:
  - 0.229
  - 0.224
  - 0.225
  test_num_spatial_crops: 1
  test_scale: 256
  train_horizontal_flip: true
  train_jitter_max: 336
  train_jitter_min: 224
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/aak
  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