ganset.txt

config/classify.json

@@ -3,12 +3,22 @@
       "name":"celeba",
       "train_file_path":"./data/trainset.txt",
       "test_file_path":"./data/testset.txt",
-      "img_path": "./data/img_align_celeba_png",
+      "img_path": "./data/img_align_celeba_png/img_align_celeba/img_align_celeba",
       "model_name":"VGG16",
       "mode":"reg",
       "n_classes":1000,
       "gpus":"0,1,2,3,4,5,6,7"
     },
+    "resnet18":{
+      "epochs":30,
+      "batch_size":64,
+      "instance":4,
+      "lr":1e-2,
+      "momentum":0.9,
+      "weight_decay":1e-4,
+      "gamma":0.2,
+      "adjust_epochs":[20, 35]
+    },
     "VGG16":{
       "epochs":50,
       "batch_size":64,
@@ -68,4 +78,4 @@
     }
   }
 
-
+