mirror of
https://github.com/FoundationVision/GLEE.git
synced 2025-06-03 14:51:17 +08:00
16 lines
396 B
Python
16 lines
396 B
Python
|
import torch
|
||
|
|
|
||
|
|
from detectron2.config import LazyCall as L
|
||
|
|
from detectron2.solver.build import get_default_optimizer_params
|
||
|
|
|
||
|
|
SGD = L(torch.optim.SGD)(
|
||
|
|
params=L(get_default_optimizer_params)(
|
||
|
|
# params.model is meant to be set to the model object, before instantiating
|
||
|
|
# the optimizer.
|
||
|
|
weight_decay_norm=0.0
|
||
|
|
),
|
||
|
|
lr=0.02,
|
||
|
|
momentum=0.9,
|
||
|
|
weight_decay=1e-4,
|
||
|
|
)
|