def test_get_optimizer_name_with_lightning_optimizer(): from pytorch_lightning.core.optimizer import LightningOptimizer adam = torch.optim.Adam(torch.nn.Linear(1, 1).parameters()) assert _get_optimizer_name(LightningOptimizer(adam)) == "Adam"
def test_get_optimizer_name(): adam = torch.optim.Adam(torch.nn.Linear(1, 1).parameters()) assert _get_optimizer_name(adam) == "Adam"