Week04 suggestion (#493)
Week04 suggestion (#493)

* fix

* add fixes for atari rooms

* Update homework_pytorch_main.ipynb

* Update homework_pytorch_main.ipynb

* Update homework_pytorch_main.ipynb
yandexdataschool/Practical_RL,yandexdataschool/Practical_RL