wb-droid's picture
Use torch save.
b43a417