未验证 提交 06e5a1bd 编写于 作者: B Bubbliiiing 提交者: GitHub

Update train_with_tensorboard.py

上级 a77bceb3
......@@ -149,9 +149,10 @@ if __name__ == "__main__":
#-------------------------------#
Use_Data_Loader = True
#------------------------------------------------------#
# 是否对损失进行归一化
# 是否对损失进行归一化,用于改变loss的大小
# 用于决定计算最终loss是除上batch_size还是除上正样本数量
#------------------------------------------------------#
normalize = True
normalize = False
#-------------------------------#
# 输入的shape大小
# 显存比较小可以使用416x416
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册