From b83cd8eba837348bc194db8c1dd12e369602e4ac Mon Sep 17 00:00:00 2001 From: Michael Foiani Date: Wed, 4 May 2022 17:37:55 -0400 Subject: got it to run, but it hangs --- main.py | 2 ++ 1 file changed, 2 insertions(+) (limited to 'main.py') diff --git a/main.py b/main.py index 063670b8..8363fcef 100644 --- a/main.py +++ b/main.py @@ -55,7 +55,9 @@ def main(): print('this is',ARGS.content) content_image = imread(ARGS.content) + content_image = np.resize(content_image, (255, 255, 3)) style_image = imread(ARGS.style) + style_image = np.resize(style_image, (255, 255, 3)) my_model = YourModel(content_image=content_image, style_image=style_image) train(my_model) -- cgit v1.2.3-70-g09d2