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.
13 lines
453 B
13 lines
453 B
###########################################################
|
|
# PARAS SETTING #
|
|
###########################################################
|
|
# Set to -1 to indicate that the parameter is the same as the pretrained model configuration
|
|
|
|
batch_size: 10
|
|
learning_rate: 0.0001 # learning rate
|
|
num_snapshots: -1
|
|
|
|
# frozen_layers should be a list
|
|
# if you don't need to freeze, set frozen_layers to []
|
|
frozen_layers: ["encoder"]
|