examples
examples copied to clipboard
GANs mnist
@kartikdutt18 @zoq I have created this new pull request coz in the earlier branch a lot of stuff has been messed up. To do:
- Document Model architecture( It's half done).
- Notebook to create samples( but this can be done after we train the model).
Please check the correctness of script, especially the Evaluate Function and model save method. @zoq If this training file is correct, can you please train the model? Right now, the model will on 657 cycles with 1 epoch per cycle. May be we can improve on this? Your thoughts.
P.S: Ignore the file path, and make file configurations, I will fix them later on.
@zoq @kartikdutt18 I forgot to mention that the split function is not working for me. Like it doesn't do anything, and the entire dataset is loaded. Can you check if it is the same for you?
@swaingotnochill @Davidportlouis I just realized that we have a notebook that shows how you can load images into an armadillo matrix with the format mlpack expects - https://github.com/mlpack/examples/blob/master/cifar10_transformation_with_pca/cifar-10-pca-cpp.ipynb. Let me know if you have any questions about the notebook.
785x42000 Dataset Loaded Train Dataset Size : (784, 42000) Validation Dataset Size : (784, 0) Training ...
I am not sure why....I did build mlpack from master...maybe I will create a new build and test it...
Check out this pull request on
See visual diffs & provide feedback on Jupyter Notebooks.
Powered by ReviewNB
The generate_image.py file is incomplete[I have yet to implement what kartik proposed]. Also the code needs a bit of cleaning. Please let me know if there is any logic error.
@zoq @kartikdutt18 I am preparing the final notebook, but it seems that it crashes very often. Will it be alright if we can have the model definition, and the explanation part on the notebook.
@zoq @kartikdutt18 Can you review the code? See if this can merge. Later, after the model is trained, we can just push the updated output. As for the notebook, I have just added the output code. The model architecture is written in comments in the mnist_gan.cpp file. Just let me know if we want to write the same model architecture in notebook too. Also, if there is anything to add, feel free to tell.
This issue has been automatically marked as stale because it has not had any recent activity. It will be closed in 7 days if no further activity occurs. Thank you for your contributions! :+1:
@swaingotnochill Is there anything else required for this PR? This addition is very nice.
@swaingotnochill Is there anything else required for this PR? This addition is very nice.
The saved model is only trained for 2 epochs...We can merge this, just that we have to train the model for this which might take some days. But, the architecture seems fine for now, so shouldn't be an issue if we add this..
@zoq do we merge this?
I will try to port the C++ example manually to the examples repo.
@geekypathak21 feel free when you have the time to port this one as well, many thanks
@shrit Sure will do :+1: