File size: 2,966 Bytes
2b12a2a |
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 133 134 |
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
|