Update train.py

train.py

@@ -40,7 +40,10 @@ if __name__ == "__main__":
     Use_Data_Loader = True
 
     model = get_ssd("train",Config["num_classes"])
-
+    
+    #-------------------------------------------#
+    #   权值文件的下载请看README
+    #-------------------------------------------#
     print('Loading weights into state dict...')
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
     model_dict = model.state_dict()