Added se_resnet to list of classification networks

JoonVan · Aug 30, 2018 · daf2906 · daf2906
1 parent ca04926
commit daf2906
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/niftynet/engine/application_factory.py b/niftynet/engine/application_factory.py
@@ -64,6 +64,7 @@
 
     # classification
     "resnet": 'niftynet.network.resnet.ResNet',
+    "se_resnet": 'niftynet.network.se_resnet.SE_ResNet',
 
     # autoencoder
     "vae": 'niftynet.network.vae.VAE'