Train the model using GPU
divyat09 opened this issue · comments
How should I change the model.py and train.py code to enable it to use GPU for training ?
paarthneekhara / text-to-image
Text to image synthesis using thought vectors
Text to image synthesis using thought vectors
divyat09 opened this issue · comments
How should I change the model.py and train.py code to enable it to use GPU for training ?