Update README.md

README.md

@@ -58,6 +58,9 @@ model = timm.create_model(
   pretrained=True,
 ).eval()
 
+# disable norm5's activation, as per torchvision's implementation
+model.features.norm5.act == nn.Identity()
+
 # get model specific transforms (normalization, resize)
 data_config = timm.data.resolve_model_data_config(model)
 transforms = timm.data.create_transform(**data_config, is_training=False)
@@ -87,6 +90,9 @@ model = timm.create_model(
   num_classes=0,
 ).eval()
 
+# disable norm5's activation, as per torchvision's implementation
+model.features.norm5.act == nn.Identity()
+
 # get model specific transforms (normalization, resize)
 data_config = timm.data.resolve_model_data_config(model)
 transforms = timm.data.create_transform(**data_config, is_training=False)