updates
This commit is contained in:
parent
5177f3e7a0
commit
b1a2735338
3
train.py
3
train.py
|
@ -63,8 +63,7 @@ def main(opt):
|
||||||
|
|
||||||
# Set optimizer
|
# Set optimizer
|
||||||
# optimizer = torch.optim.Adam(filter(lambda p: p.requires_grad, model.parameters()))
|
# optimizer = torch.optim.Adam(filter(lambda p: p.requires_grad, model.parameters()))
|
||||||
optimizer = torch.optim.SGD(filter(lambda p: p.requires_grad, model.parameters()), lr=1e-3, momentum=.9,
|
optimizer = torch.optim.SGD(filter(lambda p: p.requires_grad, model.parameters()))
|
||||||
weight_decay=5e-4, nesterov=True)
|
|
||||||
|
|
||||||
start_epoch = checkpoint['epoch'] + 1
|
start_epoch = checkpoint['epoch'] + 1
|
||||||
if checkpoint['optimizer'] is not None:
|
if checkpoint['optimizer'] is not None:
|
||||||
|
|
Loading…
Reference in New Issue