search

dmitry-vorobiev / generative-models

my playground for generative modeling
Apache License 2.0
2 stars 2 forks source link

Minor fix of GPU index #2

Closed vfdev-5 closed 4 years ago

vfdev-5 commented 4 years ago

Hi @dmitry-vorobiev

Here is a minor fix of GPU index to make it work on single device. What do you think ?

dmitry-vorobiev commented 4 years ago

Yeah, the config is pretty dirty right now. It's what I currently use for small testing.