update train resnet script

a741419244 · Nov 15, 2020 · a2e1473 · a2e1473
1 parent fa4ded9
commit a2e1473
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_classification/Test5_resnet/train.py b/pytorch_classification/Test5_resnet/train.py
@@ -21,7 +21,7 @@ def main():
                                    transforms.CenterCrop(224),
                                    transforms.ToTensor(),
                                    transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])])}
-    
+
     data_root = os.path.abspath(os.path.join(os.getcwd(), "../.."))  # get data root path
     image_path = data_root + "/data_set/flower_data/"  # flower data set path