import torch.utils.checkpoint

#4
by IANZHU - opened

Otherwise, there wii be AttributeError: module 'torch.utils' has no attribute 'checkpoint'

Ready to merge
This branch is ready to get merged automatically.
Your need to confirm your account before you can post a new comment.

Sign up or log in to comment