search

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 keyword only params #6

Closed mberr closed 2 years ago

mberr commented 2 years ago

Fix https://github.com/mberr/torch-max-mem/issues/5