Function build_optimizer

astroclip/astrodino/trainer.py:73–76 · view source on GitHub ↗

(cfg, params_groups)

Source from the content-addressed store, hash-verified

71
72
73	def build_optimizer(cfg, params_groups):
74	return torch.optim.AdamW(
75	params_groups, betas=(cfg.optim.adamw_beta1, cfg.optim.adamw_beta2)
76	)
77
78
79	def build_schedulers(cfg):

do_trainFunction · 0.85

no outgoing calls

no test coverage detected