Update deepfake_detection_test.py

vishal3477 · Jul 24, 2021 · 36646ac · 36646ac
1 parent aa67fa0
commit 36646ac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepfake_detection_test.py b/deepfake_detection_test.py
@@ -65,7 +65,7 @@
 test_loader = torch.utils.data.DataLoader(test_set,batch_size=32,shuffle =True, num_workers=1)
 
 
-model=fen.DnCNN().to(device)
+model=fen.DnCNN(num_layers=31).to(device)
 
 model_params = list(model.parameters())    
 optimizer = torch.optim.Adam(model_params, lr=opt.lr)