fixed typo in aiaccel/examples/resnet50_cifar10/user.py

aistairc · Mar 28, 2024 · 9320f4e · 9320f4e
1 parent d0d28b3
commit 9320f4e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/resnet50_cifar10/user.py b/examples/resnet50_cifar10/user.py
@@ -3,7 +3,7 @@
 import torch
 import torch.nn as nn
 import torch.optim as optim
-from torch.optim.lr_manager import CosineAnnealingLR
+from torch.optim.lr_scheduler import CosineAnnealingLR
 from torchvision import datasets, models, transforms
 
 from aiaccel.util import aiaccel