Verified Commit 3e432454 authored by Max Ehrlich's avatar Max Ehrlich
Browse files

For number format in printing progress

parent 70e704fd
......@@ -18,7 +18,7 @@ args = parser.parse_args()
spatial_accuracies = 0
jpeg_accuracies = 0
for m in range(args.models):
print('Train spatial model {}/{}'.format(m, args.models))
print('Train spatial model {}/{}'.format(m+1, args.models))
spatial_dataset = data.spatial_dataset_map[args.dataset](args.batch_size, args.data)
dataset_info = data.dataset_info[args.dataset]
spatial_model = models.SpatialResNet(dataset_info['channels'], dataset_info['classes']).to(device)
......
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