Skip to content
Snippets Groups Projects
Commit 939d07af authored by tgupta6's avatar tgupta6
Browse files

begining to make select_best_model compatible

parent 1a015d00
No related branches found
No related tags found
No related merge requests found
...@@ -48,7 +48,7 @@ def create_batch_generator(): ...@@ -48,7 +48,7 @@ def create_batch_generator():
constants.object_labels_json, constants.object_labels_json,
constants.attribute_labels_json, constants.attribute_labels_json,
constants.regions_json, constants.regions_json,
constants.genome_train_held_out_region_ids, constants.genome_val_region_ids,
constants.image_size, constants.image_size,
channels=3, channels=3,
resnet_feat_dim = constants.resnet_feat_dim, resnet_feat_dim = constants.resnet_feat_dim,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment