aak / config.yaml
kiyoonkim's picture
Upload config.yaml with huggingface_hub
a56d54f
raw
history blame
3.12 kB
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