search

Newmu / dcgan_code

Deep Convolutional Generative Adversarial Networks
MIT License
3.44k stars 694 forks source link

Training input dimensions #29

Open ch3njust1n opened 7 years ago

ch3njust1n commented 7 years ago

I'm trying to create my own dataset. What's the order of the dimensions of the input tensor? Is it (batch_size, channels, height, width), (batch_size, height, width, channels), (batch_size, height width channels, 1), or something different? I'm having a lot of trouble creating the hdf5 file. Anyone else have any luck?

Newmu commented 7 years ago

format is (batch, channel, height, width) also called bc01 in theano docs.

ch3njust1n commented 7 years ago

@Newmu If I follow the example (http://fuel.readthedocs.io/en/latest/h5py_dataset.html) for creating a HDF5 dataset, will that do the trick for training on new data or are there other things I should be aware of?