ColorTransferLibModels / NIMA /config_aesthetic_cpu.json
hpotechius's picture
Upload 6 files
76840a3 verified
Raw
History Blame Contribute Delete
430 Bytes
{
"train_env": "remote",
"docker_image": "nima-cpu",
"base_model_name": "MobileNet",
"existing_weights": null,
"n_classes": 10,
"batch_size": 96,
"epochs_train_dense": 5,
"learning_rate_dense": 0.001,
"decay_dense": 0,
"epochs_train_all": 9,
"learning_rate_all": 0.00003,
"decay_all": 0.000023,
"l2_reg": null,
"dropout_rate": 0.75,
"multiprocessing_data_load": false,
"num_workers_data_load": 1
}