Update README.md so that it no longer uses nonexisting arguments in t…

…he example scripts
DeepRank · Dec 12, 2023 · e2cc1db · e2cc1db
1 parent 58c4d68
commit e2cc1db
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -185,7 +185,7 @@ data_set = DataSet(
 
 # create the network
 model = NeuralNet(data_set,cnn_reg,model_type='3d',task='reg',
-                  cuda=False,plot=True,outdir=out)
+                  cuda=False)
 
 # change the optimizer (optional)
 model.optimizer = optim.SGD(model.net.parameters(),
@@ -233,7 +233,7 @@ data_set = DataSet(
 model = NeuralNet(data_set, cnn_reg, model_type='3d', task='reg',
                   pretrained_model="best_valid_model.pth.tar",
                   metrics_exporters=[OutputExporter(out)],
-                  cuda=False, plot=True, outdir=out)
+                  cuda=False)
 
 # change the optimizer (optional)
 model.optimizer = optim.SGD(model.net.parameters(),