search

bdzyubak / torch-control

A top-level repo for evaluating natively available models
MIT License
2 stars 0 forks source link

Adapt GPU memory target to system memory #8

Closed bdzyubak closed 3 months ago

bdzyubak commented 3 months ago

Currently, the target GPU memory footprint hardcoded and needs to be overridden via inputs.

Add feature to adapt this to system memory automatically.

bdzyubak commented 3 months ago

Added utility to check GPU memory and calls in nnUnet to use it instead of hardcoded values.