search

andrewgordonwilson / bayesgan

Tensorflow code for the Bayesian GAN (https://arxiv.org/abs/1705.09558) (NIPS 2017)
Other
1.02k stars 176 forks source link

Custom data #1

Closed usptact closed 6 years ago

usptact commented 6 years ago

To learn on custom data (not images), are the shapes hard-coded to 8x8 or I can change them? I essentially have flat vectors.

ysaatchi commented 6 years ago

The models certainly expect images for now unfortunately, shouldn't be too hard to hack a version which takes arbitrary network structures (ideally in pytorch ;))