File size: 2,938 Bytes
9784b0c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 |
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-05
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
|