callbacks:
  hf_uploader:
    _target_: mlproject.callbacks.UploadCheckpointsToHuggingFace
    repo_name: aag
    repo_owner: kiyoonkim
code_dir: /home/kiyoon/project/mlproject
current_experiment_dir: /disk/scratch2/experiments/aag
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: aag
freeze_backbone: false
hf_repo_dir: /disk/scratch2/experiments/aag/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/aag/repo
  experiment_name: aag
  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.adamw.AdamW
  amsgrad: false
  betas:
  - 0.9
  - 0.999
  capturable: false
  differentiable: false
  eps: 1.0e-08
  foreach: null
  fused: null
  lr: 1.0e-06
  maximize: false
  params: ???
  weight_decay: 1.0e-05
print_config: false
repo_path: kiyoonkim/aag
resume: false
resume_from_checkpoint: null
root_experiment_dir: /disk/scratch2/experiments
scheduler:
  _target_: torch.optim.lr_scheduler.MultiStepLR
  gamma: 0.1
  last_epoch: -1
  milestones:
  - 2000
  - 4000
  - 6000
  - 8000
  optimizer: ???
  verbose: false
seed: 42
test: false
torch_dataset:
  _target_: mlproject.torch_datasets.gulp_sparsesample.build_gulp_dataset
  data_dir: /disk/scratch_fast1/datasets
  dataset_name: hmdb51-gulprgb
  sets_to_include: null
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/aag
  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