Skip to content
Snippets Groups Projects
Commit aa2523a9 authored by Yandi's avatar Yandi
Browse files

[Run] Python script for optimizers

parent e30f18d4
No related branches found
No related tags found
1 merge request!1Master into main
import torch.optim
def optimizer(cfg):
result = {"Adam" : torch.optim.Adam(model.parameters())}
return result[cfg["Optimizer"]]
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment