From 2d24ba75ab94368d00a67f840d34927ec715e827 Mon Sep 17 00:00:00 2001 From: Bubbliiiing <47347516+bubbliiiing@users.noreply.github.com> Date: Sat, 23 Jan 2021 12:05:39 +0800 Subject: [PATCH] Update train.py --- train.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index 36d50bf..fb3f156 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参数 -- GitLab