search

zhanghang1989 / PyTorch-Multi-Style-Transfer

Neural Style and MSG-Net
http://hangzh.com/PyTorch-Style-Transfer/
MIT License
976 stars 207 forks source link

Establish Docker directory #44

Closed ss32 closed 3 years ago

ss32 commented 3 years ago

What: Establishes a Docker directory with Dockerfile and run script

Why: The original repo was written for an outdated version of PyTorch, which makes it hard to run on modern systems without conflicting with updated versions of the dependencies.

Build the container with 

cd Docker
docker build -t style-transfer .
zhanghang1989 commented 3 years ago

Thanks for the contribution!