diff --git a/tests/methods/style_transfer/test_ccpl.py b/tests/methods/style_transfer/test_ccpl.py index bcff007..f9e9bb2 100644 --- 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)