search

facebookarchive / caffe2

Caffe2 is a lightweight, modular, and scalable deep learning framework.
https://caffe2.ai
Apache License 2.0
8.42k stars 1.95k forks source link

GANs in caffe2 #451

Open peterneher opened 7 years ago

peterneher commented 7 years ago

Is it possible to implement GANs in Caffe2? It was not possible in Caffe if I recall correctly.

Cheers, Peter

peterneher commented 7 years ago

I'm sure someone knows if this is possible :) Maybe @bwasti ?

yxchng commented 6 years ago

@peterneher Have you figured out if it is possible and if it is good to use caffe2 for that?

edubois commented 6 years ago

Any news on this topic? It's very important...

leon532 commented 6 years ago

https://github.com/caffe2/caffe2/commit/fd5fbca0a814305012435d9571bdf2183a9e5b7e

I found this. But i am still not sure how to use the AlternateLearnigRate to implement GAN.

eloi-loomai commented 6 years ago

I confirm it is possible, was able to implement a dcgan, boundary eq. gan is harder to implement though.