Added B4 and B5 models

Author: lukemelas

efficientnet_pytorch/model.py

@@ -158,7 +158,7 @@ def extract_features(self, inputs):
             drop_connect_rate = self._global_params.drop_connect_rate
             if drop_connect_rate:
                 drop_connect_rate *= float(idx) / len(self._blocks)
-            x = block(x) # , drop_connect_rate) # see https://github.com/tensorflow/tpu/issues/381
+            x = block(x, drop_connect_rate) 
 
         return x
 

setup.py

@@ -18,7 +18,7 @@
 EMAIL = '[email protected]'
 AUTHOR = 'Luke'
 REQUIRES_PYTHON = '>=3.5.0'
-VERSION = '0.1.0'
+VERSION = '0.2.0'
 
 # What packages are required for this module to be executed?
 REQUIRED = [