doc: Add doc about deprecation

owkin · Jan 4, 2024 · ea063af · ea063af
1 parent c2673d3
commit ea063af
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/flamby/datasets/fed_camelyon16/dataset_creation_scripts/tiling_slides.py b/flamby/datasets/fed_camelyon16/dataset_creation_scripts/tiling_slides.py
@@ -178,6 +178,8 @@ def main(
         tissue_percent=60,
     )
 
+    # syntax with pretrained=True is now deprecated
+    # https://pytorch.org/vision/stable/models.html#initializing-pre-trained-models
     resnet_weights = models.ResNet50_Weights.IMAGENET1K_V1
     net = models.resnet50(weights=resnet_weights)
     net.fc = Identity()