mxochicale
commented
1 year ago 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'
Zhanxiang (Sean) Sun @seansunn made great implementation of
style-swin-ganhere 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!