mberr / torch-max-mem

Decorators for maximizing memory utilization with PyTorch & CUDA
https://torch-max-mem.readthedocs.io/en/latest/
MIT License
14 stars 0 forks source link

Fix annotation check #13

Closed mberr closed 1 year ago

mberr commented 1 year ago

Fix #12