Improve typing - Githubissues

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

Improve typing #18

Closed mberr closed 1 year ago

mberr commented 1 year ago

use ParamSpec to annotate wrapped callable
(some cleanup of requirements)