Save checkpoints in .pkl format

lucidrains / lightweight-gan

Implementation of 'lightweight' GAN, proposed in ICLR 2021, in Pytorch. High resolution image generations that can be trained within a day or two

MIT License

1.62k stars 220 forks source link

Open JaquesM opened 2 years ago

JaquesM commented 2 years ago

Is there anyway to save the checkpoints in .pkl format, instead of .pt, or to convert the .pt file into a .pkl file?