tests/style_transfer/test_ccpl: update tests for ccpl

deel-ai · Jan 30, 2024 · 97ef52b · 97ef52b
1 parent 1e7e383
commit 97ef52b
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tests/methods/style_transfer/test_ccpl.py b/tests/methods/style_transfer/test_ccpl.py
@@ -50,9 +50,7 @@ def test_ccpl_generate_output():
     batch_size = [10 , 20]
     channel = [3 , 3]
     height, width = (256 , 256)
-    vgg_path = ('/home/vuong.nguyen/vuong/augmentare/augmentare/'
-                'methods/style_transfer/model/vgg_normalised_ccpl.pth')
-
+    vgg_path = None
     for i in range(2):
         model = CCPL(training_mode= "art", vgg_path=vgg_path, device=device)