Skip to content
Snippets Groups Projects
Commit 2eb29787 authored by Yandi's avatar Yandi
Browse files

fixing type (missing r)

parent 7a1a1f97
No related branches found
No related tags found
1 merge request!1Master into main
...@@ -95,8 +95,8 @@ if __name__ == "__main__": ...@@ -95,8 +95,8 @@ if __name__ == "__main__":
valid_ratio, valid_ratio,
overwrite_index = True, overwrite_index = True,
max_num_samples=max_num_samples, max_num_samples=max_num_samples,
train_transform= eval(dataset_transfom), train_transform= eval(dataset_transform),
valid_transform=eval(dataset_transfom) valid_transform=eval(dataset_transform)
) )
if use_cuda : if use_cuda :
...@@ -135,4 +135,4 @@ if __name__ == "__main__": ...@@ -135,4 +135,4 @@ if __name__ == "__main__":
utils.write_summary(logdir, network, optimizer, val_loss) utils.write_summary(logdir, network, optimizer, val_loss)
create_submission.create_submission(network, eval(dataset_transfom), device, rootDir, logdir) create_submission.create_submission(network, eval(dataset_transform), device, rootDir, logdir)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment