Skip to content

KarthikGanesan88/WCA-Net

 
 

Repository files navigation

WCA-Net

How to train a new model


python run.py train <path to configuration file>

Configuration files

  • m0: Vanilla model (backbone + classification layer)
  • m1: Stochastic isotropic model, trained with WCA
  • m2: Stochastic anisotropic model, trained with WCA
  • m3: Stochastic anisotropic model, trained with WCA, adversarial training

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 81.6%
  • Cuda 13.3%
  • C++ 4.4%
  • Shell 0.7%