diff --git a/train.py b/train.py index 39b0a0056fcef46f5cc782472edabcd906a57634..03b3410ccc8d5ba1e97f44257d431f25ab0d460d 100644 --- a/train.py +++ b/train.py @@ -139,9 +139,10 @@ if __name__ == "__main__": #-------------------------------# Use_Data_Loader = True #------------------------------------------------------# - # 是否对损失进行归一化 + # 是否对损失进行归一化,用于改变loss的大小 + # 用于决定计算最终loss是除上batch_size还是除上正样本数量 #------------------------------------------------------# - normalize = True + normalize = False #-------------------------------# # 输入的shape大小 # 显存比较小可以使用416x416