Adding convnets and getting onto the GPU

wellecks / vaes

Variational Autoencoders & Normalizing Flows Project

19 stars 10 forks source link

Adding convnets and getting onto the GPU #4

Open isaachenrion opened 7 years ago

isaachenrion commented 7 years ago

We should really be using convnets... My computer can't really handle them though. I have a LeNet-style encoder training (and doing a good job!) but 20x slower than the single hidden layer encoder.