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

Update train.py

上级 e70faeb9
......@@ -171,7 +171,7 @@ if __name__ == "__main__":
lr = 1e-3
Batch_size = 4
Init_Epoch = 0
Freeze_Epoch = 25
Freeze_Epoch = 50
optimizer = optim.Adam(net.parameters(),lr,weight_decay=5e-4)
lr_scheduler = optim.lr_scheduler.ReduceLROnPlateau(optimizer, factor=0.5, patience=2, verbose=True)
......@@ -207,8 +207,8 @@ if __name__ == "__main__":
#--------------------------------------------#
lr = 1e-4
Batch_size = 4
Freeze_Epoch = 25
Unfreeze_Epoch = 50
Freeze_Epoch = 50
Unfreeze_Epoch = 100
optimizer = optim.Adam(net.parameters(),lr,weight_decay=5e-4)
lr_scheduler = optim.lr_scheduler.ReduceLROnPlateau(optimizer, factor=0.5, patience=2, verbose=True)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册