Update test.py

DeepRank · Dec 20, 2021 · b5afad6 · b5afad6
1 parent eae6637
commit b5afad6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paper_pretrained_models/3DeepFace/test.py b/paper_pretrained_models/3DeepFace/test.py
@@ -42,7 +42,7 @@
 # You need to add path for the dataset
 database = glob.glob('*hdf5')
 
-model = NeuralNet(database,cnn3d_class,pretrained_model='best_model.pt', outdir=outpath)
+model = NeuralNet(database,cnn3d_class,pretrained_model='best_model.pt', outdir=./)
 
 # test the pre-trained model on new data
 model.test()