mirror of https://github.com/FoundationVision/GLEE
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,
|
|
)
|