File size: 3,118 Bytes
a56d54f |
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 135 136 137 138 139 140 141 142 |
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
|