Replicating and refactoring `style-swin-gan`

[mxochicale]

Zhanxiang (Sean) Sun @seansunn made great implementation of style-swin-gan here https://github.com/budai4medtech/midl2023/tree/main/notebooks/style-swin-gan which will be great to add in the lib to improve reproducibility and potentially refactor few bits!

[mxochicale]

Some issues to solve:

• [ ] CUDA memory > https://github.com/budai4medtech/xfetus/blob/32-style-swin-gan/examples/style-swin-gan/styleswin_tt.ipynb

  OutOfMemoryError: CUDA out of memory. Tried to allocate 192.00 MiB (GPU 0; 7.79 GiB total capacity; 6.42 GiB already allocated; 8.50 MiB free; 6.58 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF

• [ ] checkpoint >https://github.com/budai4medtech/xfetus/blob/32-style-swin-gan/examples/style-swin-gan/styleswin_tc.ipynb

  FileNotFoundError: [Errno 2] No such file or directory: 'data/saved_tt_256/d_500.pt'