Commit a9efc259 authored by Merlijn Wajer's avatar Merlijn Wajer
Browse files

Stub files for dhSegment training

parent 844270a1
{
"training_params" : {
"learning_rate": 5e-5,
"batch_size": 1,
"make_patches": false,
"training_margin" : 0,
"n_epochs": 30,
"data_augmentation" : true,
"data_augmentation_max_rotation" : 0.2,
"data_augmentation_max_scaling" : 0.2,
"data_augmentation_flip_lr": true,
"data_augmentation_flip_ud": true,
"data_augmentation_color": false,
"evaluate_every_epoch" : 5
},
"pretrained_model_name" : "resnet50",
"prediction_type": "CLASSIFICATION",
"train_data" : "archive_books/dataset/train",
"eval_data" : "archive_books/dataset/eval",
"classes_file" : "archive_books/dataset/classes.txt",
"model_output_dir" : "archive_books/book_model3",
"gpu" : "0"
}
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment