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
23 lines
329 B
23 lines
329 B
3 years ago
|
device:
|
||
|
|
device_str: null
|
||
|
|
n_gpu: -1
|
||
|
|
sync_bn: true
|
||
|
|
metrics:
|
||
|
|
metric: Accuracy
|
||
|
|
train:
|
||
|
|
batch_size: 16
|
||
|
|
learning:
|
||
|
|
optimizer:
|
||
|
|
name_: SGD
|
||
|
|
lr: 0.03
|
||
|
|
momentum: 0.001
|
||
|
|
nesterov: 111
|
||
|
|
loss:
|
||
|
|
name_: CrossEntropyLoss
|
||
|
|
label_smoothing: 0.1
|
||
|
|
#learning:
|
||
|
|
# optimizer:
|
||
|
|
# name_: Adam
|
||
|
|
# lr: 0.02
|
||
|
|
# eps: 0.001
|