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
42 lines
809 B
42 lines
809 B
callback:
|
|
early_stopping:
|
|
mode: max
|
|
monitor: eval_epoch_metric
|
|
patience: 2
|
|
model_checkpoint:
|
|
every_n_epoch: 2
|
|
tensorboard:
|
|
comment: ''
|
|
log_dir: 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:
|
|
logging_dir: null
|
|
logging_strategy: steps
|
|
print_steps: null
|
|
metrics:
|
|
metric: Accuracy
|
|
train:
|
|
batch_size: 16
|
|
dataloader_drop_last: false
|
|
dataloader_num_workers: 0
|
|
dataloader_pin_memory: true
|
|
epoch_num: 3
|
|
eval_steps: null
|
|
eval_strategy: epoch
|
|
freeze_bn: true
|
|
load_best_model_at_end: false
|
|
output_dir: ./output_dir
|
|
overwrite_output_dir: true
|
|
resume_from_checkpoint: null
|
|
seed: 42
|
|
val_batch_size: -1
|