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 default value handling #8

Closed mberr closed 2 years ago

mberr commented 2 years ago

This PR fixes handling of default values for the parameters to optimize.