Upload config.yaml with huggingface_hub

config.yaml

@@ -0,0 +1,133 @@
+callbacks:
+  hf_uploader:
+    _target_: mlproject.callbacks.UploadCheckpointsToHuggingFace
+    repo_name: aae
+    repo_owner: kiyoonkim
+code_dir: /home/kiyoon/project/mlproject
+current_experiment_dir: /disk/scratch2/experiments/aae
+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.data.build_gulp_dataset
+  data_dir: /disk/scratch_fast1/datasets
+  dataset_name: hmdb51-gulprgb
+  sets_to_include: null
+download_checkpoint_with_name: null
+download_latest: true
+eval_batch_size: 8
+exp_name: aae
+freeze_backbone: false
+hf_repo_dir: /disk/scratch2/experiments/aae/repo
+hf_username: kiyoonkim
+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/aae/repo
+  experiment_name: aae
+  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.build_tsm_model
+  model_name: tsm_resnet50
+  num_classes: 51
+  num_frames: 8
+  pretrained: kinetics400
+num_workers: 4
+optimizer:
+  _partial_: true
+  _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-05
+  maximize: false
+  weight_decay: 1.0e-05
+print_config: false
+repo_path: kiyoonkim/aae
+resume: false
+resume_from_checkpoint: null
+root_experiment_dir: /disk/scratch2/experiments
+scheduler:
+  _partial_: true
+  _target_: timm.scheduler.cosine_lr.CosineLRScheduler
+  cycle_decay: 1.0
+  cycle_limit: 1
+  cycle_mul: 1.0
+  initialize: true
+  k_decay: 1.0
+  lr_min: 0.0
+  noise_pct: 0.67
+  noise_range_t: null
+  noise_seed: 42
+  noise_std: 1.0
+  t_in_epochs: true
+  warmup_lr_init: 0
+  warmup_prefix: false
+  warmup_t: 0
+seed: 42
+test: false
+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/aae
+  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