refactor(train): 修改数据读取路径

Author: zjZSTU

py/lib/train.py

@@ -110,7 +110,7 @@ def train_model(data_loader, model, criterion, optimizer, lr_scheduler, num_epoc
     device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
     # device = "cpu"
 
-    data_loader = load_data('../data/training_images', cate_list, S=S, B=B, C=C)
+    data_loader = load_data('../data/location_dataset', cate_list, S=S, B=B, C=C)
     # print(len(data_loader))
 
     model = YOLO_v1(S=S, B=B, C=C)