This is an implementation of a simple Generative Adversarial Network for semi-supervised learning on PyTorch. Originally I studied the subject by Udacity Deep Learning Foundation Nanodegree, but the original implementation on TensorFlow was a bit wierd for me. PyTorch provides more clear way to implement such a network. I achieved the accuracy of 70% on the test data set after training for 80 epochs. This accuracy corresponds to that one achieved by the TensorFlow implementation.
This implementation is only an entry point and definitely could and should be improved to get the much better accuracy. It is partially inspired by this work.
In the repository you can find the data loader, model and solver separated in different Python files and a Jupyter notebook that contains all-in-one if you want to try things faster.
-
Notifications
You must be signed in to change notification settings - Fork 6
vbezgachev/semi-supervised-gan
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
PyTorch implementation of semi-supervised GAN
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published