Open leomrocha opened 4 years ago
I made a few changes to make it work under python3 and pytorch v1, I tried it and seems to work correctly and finishes in 52 minutes (in my setup of python3.7 and torch v1.3.1 gpu). I also made some aesthetic changes to comments (typos for example)
I made a few changes to make it work under python3 and pytorch v1, I tried it and seems to work correctly and finishes in 52 minutes (in my setup of python3.7 and torch v1.3.1 gpu). I also made some aesthetic changes to comments (typos for example)