Requirements not specified

[hno2lr-xcdx]

The readme contains information how to run the python scripts and the notebook.

• for installing the scripts: Which requirements are needed?
• hint: this Pull request is hinting to some requirements
• Hardware/pypi packages?

I ran the scripts on a laptop with 4GB GPU size.
But:

• only the generate_image_pairs is working, the other script style_transfer_folder is throwing a CUDA OutOfMemoryError.
• jupyter notebook is throwing the same CUDA error.

Thanks in advance for a short feedback.

[onion-liu]

The script style_transfer_folder.py requires more GPU memory than generate_image_pairs.py because it loads another model (PSPEncoder). Perhaps you need another GPU with more memory to run style_transfer_folder.py

You can also split style_transfer_folder.py into two steps:
Step 1: calculate sample_in with psp_encoder and save sample_in into a .npy file.
Step 2: generate img_out_ten from sample_in with the model g_ema to get the final result.

[hno2lr-xcdx]

Ok, thanks for the feedback.

Is it possible to mitigate the problem by using a smaller version of the PSPEncoder model?

[onion-liu]

Yes. But you need to train the new version of the PSPEncoder by yourself.