aug_model:
  _target_: mlproject.aug_models.augmentation.VideoAugmentation
callbacks:
  hf_uploader:
    _target_: mlproject.callbacks.UploadCheckpointsToHuggingFace
    repo_name: aa1
    repo_owner: kiyoonkim
code_dir: /home/kiyoon/project/mlproject
current_experiment_dir: /disk/scratch2/experiments/aa1
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: aa1
freeze_backbone: false
hf_repo_dir: /disk/scratch2/experiments/aa1/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/aa1/repo
  experiment_name: aa1
  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/aa1
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/hmdb51
  dataset_name: hmdb51-gulprgb
  ensure_installed: true
  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/aa1
  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