from torch.optim.lr_scheduler import LambdaLR, _LRScheduler

Beckschen / 3D-TransUNet

This is the official repository for the paper "3D TransUNet: Advancing Medical Image Segmentation through Vision Transformers"

Apache License 2.0

182 stars 10 forks source link

from torch.optim.lr_scheduler import LambdaLR, _LRScheduler #11

Open usst-changbo opened 10 months ago

usst-changbo commented 10 months ago

When using lr_schedule, from torch.optim.lr_scheduler import LambdaLR, _LRScheduler cannot be imported because it is an internal class. What should I do