Commit 634db1b4 authored by Fernando Perez-Garcia's avatar Fernando Perez-Garcia

Use same window size for inference

parent c4aa2a64
......@@ -24,13 +24,14 @@ for net in nets:
string = '-'.join(inputs)
model_name = f'highres3dnet_{net}_{string}'
print(f'Submitting {model_name}...')
print(f'Creating {model_name}...')
model_dir = models_dir / model_name
model = Model(model_dir=model_dir, inputs=inputs)
model.set_images_and_labels_paths()
model.make_csv_files(split_type='subject')
model.config_all()
model.config_input_data(spatial_window_size=96)
model.config_inference(spatial_window_size=96)
model.config_network(network_name=network)
model.write_config_file()
......@@ -38,4 +39,5 @@ for net in nets:
job.write()
if len(sys.argv) > 1 and sys.argv[1] == '--submit':
print(f'Submitting {model_name}...')
job.submit()
Markdown is supported
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