update to version-0.1.1

Author: spytensor

config.py

@@ -1,28 +1,24 @@
-# -*- coding: utf-8 -*-
-# @Time    : 2018/7/31 09:41
-# @Author  : Spytensor
-# @File    : config.py
-# @Email   : zhuchaojie@buaa.edu.cn
-#====================================================
-#               定义所需要的一些参数                     
-#====================================================
-from datetime import datetime
 class DefaultConfigs(object):
-    #1.str类参数
-    data_path = "/data2/dockspace_zcj/traffic-sign/train/"
-    test_path = "/data2/dockspace_zcj/traffic-sign/test/"
-    model_name = "resnet152"
-    weights_path = "./logs/{}_params.pkl".format(model_name)
-    predict_path = "./results/{}".format("%s-lgb_two_three_features"%(datetime.now().strftime("%Y%m%d-%H%M%S")))
-    #2.数字类特征
-    img_size = 224
-    channels = 3
-    epochs = 120
-    batch_size = 32
-    lr = 1e-4
-    weight_decay = 1e-4
-    print_freq = 20
+    #1.string parameters
+    train_data = "../data/train/"
+    test_data = ""
+    val_data = "../data/val/"
+    model_name = "resnet50"
+    weights = "./checkpoints/"
+    best_models = weights + "best_model/"
+    submit = "./submit/"
+    logs = "./logs/"
+    gpus = "1"
+
+    #2.numeric parameters
+    epochs = 40
+    batch_size = 4
+    img_height = 224
+    img_weight = 224
     num_classes = 62
-    gpu = "3"
-config = DefaultConfigs()
+    seed = 888
+    lr = 1e-3
+    lr_decay = 1e-4
+    weight_decay = 1e-4
 
+config = DefaultConfigs()