towhee
/
resnet-image-embedding
copied
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Readme
Files and versions
37 lines
697 B
37 lines
697 B
callback:
|
|
early_stopping:
|
|
mode: max
|
|
monitor: eval_epoch_metric
|
|
patience: 4
|
|
model_checkpoint:
|
|
every_n_epoch: 1
|
|
tensorboard: null
|
|
device:
|
|
device_str: null
|
|
n_gpu: -1
|
|
sync_bn: false
|
|
learning:
|
|
loss: CrossEntropyLoss
|
|
lr: 5.0e-05
|
|
lr_scheduler_type: linear
|
|
optimizer: Adam
|
|
warmup_ratio: 0.0
|
|
warmup_steps: 0
|
|
logging:
|
|
print_steps: 1
|
|
metrics:
|
|
metric: Accuracy
|
|
train:
|
|
batch_size: 5
|
|
dataloader_drop_last: true
|
|
dataloader_num_workers: -1
|
|
dataloader_pin_memory: true
|
|
epoch_num: 3
|
|
eval_steps: null
|
|
eval_strategy: epoch
|
|
freeze_bn: false
|
|
load_best_model_at_end: false
|
|
output_dir: my_output
|
|
overwrite_output_dir: true
|
|
seed: 42
|
|
val_batch_size: -1
|