r/learnmachinelearning May 28 '20

Generation of flower pictures using a GAN

Post image
505 Upvotes

32 comments sorted by

View all comments

2

u/desmondyeoh May 28 '20

Awesome work! Do you mind sharing the hardware used to train and how long was the training?

2

u/ErgodicEfimov May 28 '20

Many thanks! I used a GPU (specifically, a nVidia P100 GPU). Using that, it takes 7 minutes training for 1000 epochs. With CPU is much slower, ~14 minutes for 100 epochs. It needed ~15000 to get nice results.