revert format

deepjavalibrary · Oct 12, 2022 · a13f059 · a13f059
1 parent 3aa7b76
commit a13f059
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/src/main/java/ai/djl/examples/training/transferlearning/TransferFreshFruit.java b/examples/src/main/java/ai/djl/examples/training/transferlearning/TransferFreshFruit.java
@@ -68,7 +68,7 @@ public static TrainingResult runExample(String[] args)
         }
 
         // Also available at
-        // "https://mlrepo.djl.ai/model/cv/image_classification/ai/djl/pytorch/resnet18_embedding/0.1/traced_resnet18_embedding.pt.gz";
+        // "https://mlrepo.djl.ai/model/cv/image_classification/ai/djl/pytorch/resnet18_embedding/0.0.1/traced_resnet18_embedding.pt.gz";
         String modelUrls = "/Users/fenkexin/Desktop/transferDJL/code/base_nw.pt";
         Criteria<NDList, NDList> criteria =
                 Criteria.builder()
@@ -137,11 +137,11 @@ private static ImageFolder getData(String subfolderName, String fruit, int batch
         // href="https://www.kaggle.com/datasets/sriramr/fruits-fresh-and-rotten-for-classification">https://www.kaggle.com/datasets/sriramr/fruits-fresh-and-rotten-for-classification</a>
         String folderUrl = "/Users/fenkexin/Desktop/transferDJL/code/data/" + fruit;
         String subfolder = "/" + subfolderName + "/";
-        Repository repositoryTrain =
+        Repository repository =
                 Repository.newInstance("banana", Paths.get(folderUrl + subfolder));
         ImageFolder dataset =
                 ImageFolder.builder()
-                        .setRepository(repositoryTrain)
+                        .setRepository(repository)
                         .addTransform(new ToTensor())
                         .addTransform(new Transpose(1, 2, 0))
                         .addTransform(new Resize(224, 224))