Upload config.yaml with huggingface_hub

config.yaml

@@ -0,0 +1,141 @@
+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