8359bb1
1
2
3
4
5
6
import torch def configure_optimizers(model, lr): adam = torch.optim.Adam(model.parameters(), lr=lr) return adam
