Skip to content

Y
yolov4-pytorch

last_winter / yolov4-pytorch
与 Fork 源项目一致

从无法访问的项目Fork

通知 7
Star 0
Fork 0
Y
yolov4-pytorch

前往新版Gitcode，体验更适合开发者的 AI 搜索 >>

未验证 提交 197bb35e 编写于 作者: B Bubbliiiing 提交者： GitHub
浏览文件
操作

Update train.py

上级 ec841321
隐藏空白更改
内联 并排
Showing with 3 addition and 3 deletion
train.py
浏览文件 @ 197bb35e
...@@ -190,7 +190,7 @@ if __name__ == "__main__": ...@@ -190,7 +190,7 @@ if __name__ == "__main__":
lr = 1e-3 lr = 1e-3
Batch_size = 4 Batch_size = 4
Init_Epoch = 0 Init_Epoch = 0
Freeze_Epoch = 25 Freeze_Epoch = 50
optimizer = optim.Adam(net.parameters(),lr,weight_decay=5e-4) optimizer = optim.Adam(net.parameters(),lr,weight_decay=5e-4)
if Cosine_lr: if Cosine_lr:
...@@ -226,8 +226,8 @@ if __name__ == "__main__": ...@@ -226,8 +226,8 @@ if __name__ == "__main__":
if True: if True:
lr = 1e-4 lr = 1e-4
Batch_size = 2 Batch_size = 2
Freeze_Epoch = 25 Freeze_Epoch = 50
Unfreeze_Epoch = 50 Unfreeze_Epoch = 100
optimizer = optim.Adam(net.parameters(),lr,weight_decay=5e-4) optimizer = optim.Adam(net.parameters(),lr,weight_decay=5e-4)
if Cosine_lr: if Cosine_lr:
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或