ColorTransferLibModels / NIMA /config_technical_gpu.json
hpotechius's picture
Upload 6 files
76840a3 verified
Raw
History Blame Contribute Delete
483 Bytes
{
"train_env": "TID2013",
"s3_bucket": "ds-hotel-image-assessment",
"docker_image": "idealo/nima-gpu",
"base_model_name": "MobileNet",
"existing_weights": null,
"n_classes": 10,
"batch_size": 64,
"epochs_train_dense": 25,
"learning_rate_dense": 0.001,
"decay_dense": 0,
"epochs_train_all": 25,
"learning_rate_all": 0.0000003,
"decay_all": 0,
"dropout_rate": 0.75,
"multiprocessing_data_load": true,
"num_workers_data_load": 8,
"img_format": "bmp"
}