Skip to content

doducthao/gan-margin

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 Cannot retrieve latest commit at this time.

History

39 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

GAN-SSL

  • RMCos-SSL
  • RLMSoftmax-SSL
  • RMARC-SSL

MNIST

cd to mnist/ and run

  • bash retrain_margingan_mnist.sh

to retrain marginGAN or

  • bash multiple_gpu.sh

to train gan-ssl.

CIFAR10

cd to further/ and run

  1. bash data-local/bin/prepare_cifar10.sh (prepare cifar10 dataset)

    • bash retrain_margingan_cifar10.sh

    to retrain marginGAN or

    • bash multiple_gpu.sh

    to train gan-ssl.

Changing --mode in .sh file [rmcos, rlmsoftmax, rmarc, margingan] to play with losses

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published