Skip to content

Commit

Permalink
Add different return code for CUDA exception
Browse files Browse the repository at this point in the history
  • Loading branch information
gordicaleksa committed Aug 7, 2020
1 parent 06911b1 commit 2c21b1b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion stylization_script.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@ def stylize_static_image(inference_config):
except Exception as e:
print(e)
print(f'Consider making the batch_size (current = {inference_config["batch_size"]} images) or img_width (current = {inference_config["img_width"]} px) smaller')
exit(-1)
exit(1)

else: # do stylization for a single image
content_img_path = os.path.join(inference_config['content_images_path'], inference_config['content_input'])
Expand Down

0 comments on commit 2c21b1b

Please sign in to comment.