facebookresearch / dora

Dora is an experiment management framework. It expresses grid searches as pure python files as part of your repo. It identifies experiments with a unique hash signature. Scale up to hundreds of experiments without losing your sanity.
MIT License
269 stars 24 forks source link

Add support for explicitly setting no mem_per_gpu value #22

Closed JadeCopet closed 2 years ago

JadeCopet commented 2 years ago

Add support for explicitly specifying no mem_per_gpu

This PR allows users to specify mem_per_gpu=None for Slurm. This has been tested specifying it through the yaml config, in the grid launcher.