diff --git a/train.py b/train.py index 36d50bff21dcde29bf377fc51f921cc81b8a734b..fb3f156c0a44c76e339f7e7c13c927fa66f92311 100644 --- a/train.py +++ b/train.py @@ -122,9 +122,10 @@ if __name__ == "__main__": #-------------------------------# Use_Data_Loader = True #------------------------------------------------------# - # 是否对损失进行归一化 + # 是否对损失进行归一化,用于改变loss的大小 + # 用于决定计算最终loss是除上batch_size还是除上正样本数量 #------------------------------------------------------# - normalize = True + normalize = False #------------------------------------------------------# # 创建yolo模型 # 训练前一定要修改Config里面的classes参数