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
25 lines
377 B
25 lines
377 B
device:
|
|
device_str: null
|
|
n_gpu: -1
|
|
sync_bn: true
|
|
metrics:
|
|
metric: Accuracy
|
|
train:
|
|
batch_size: 32
|
|
overwrite_output_dir: true
|
|
epoch_num: 2
|
|
learning:
|
|
optimizer:
|
|
name_: SGD
|
|
lr: 0.04
|
|
momentum: 0.001
|
|
loss:
|
|
name_: CrossEntropyLoss
|
|
ignore_index: -1
|
|
logging:
|
|
print_steps: 2
|
|
#learning:
|
|
# optimizer:
|
|
# name_: Adam
|
|
# lr: 0.02
|
|
# eps: 0.001
|