added best config model

Author: zabir-nabil

__pycache__/config.cpython-37.pyc

@@ -71,3 +71,32 @@ def __init__(self, model_type):
             self.attn_filters = [32, 64, 128] # attn_filters and cnn_dim should be same
 
             self.n_attn_layers = [1, 2, 3]
+
+        elif model_type == "attn_train_best_config":
+            # ablation study
+            self.seed = 1997
+            self.data_folder = '..' # .. one level up
+            self.ct_tab_feature_csv = 'train_data_ct_tab.csv' # some extra features
+            self.strip_ct = .15 # strip this amount of ct slices before randomly choosing
+            self.n_tab = 5 # number of tabular features used
+
+            # self.cnn_dim = 32 # compressed cnn feature dim
+
+            self.fc_dim = [32]
+
+            # select which models to train
+            self.train_models = [ 'resnet18' , 'resnet34', 'resnet50', 'resnet101', 'resnet152', 'resnext50', 'resnext101', 'efnb0', 'efnb1', 'efnb2', 'efnb3', 'efnb4' ]  
+            # train 1 : 'resnet18' , 'resnet34', 'resnet50', 'resnet101', 'resnet152', 'resnext50'
+            # train 2 : 'resnext101', 'efnb0', 'efnb1', 'efnb2', 'efnb3', 'efnb4'
+
+            self.gpu_index = 0
+            self.num_workers = 0 # 0 for bug fix/docker
+            self.results_dir = "results_attn_bc"
+            self.nfold = 5
+            self.n_epochs = 40
+            self.batch_size = 10
+            self.final_lr = 0.0002
+
+            self.attn_filters = [64] # attn_filters and cnn_dim should be same
+
+            self.n_attn_layers = [1]

config.py

@@ -130,3 +130,5 @@ widgetsnbextension==3.5.1
 wrapt==1.12.1
 zipp==3.1.0
 # kaggle
+ptflops
+thop