From daf29063f81b2e78320a07d6f0d58400510043bf Mon Sep 17 00:00:00 2001
From: Pritesh Mehta <rmappm0@ucl.ac.uk>
Date: Thu, 30 Aug 2018 14:47:20 +0100
Subject: [PATCH] Added se_resnet to list of classification networks

---
 niftynet/engine/application_factory.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/niftynet/engine/application_factory.py b/niftynet/engine/application_factory.py
index 3f6beeb3..8c7b6cd5 100755
--- 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'