Repository metrics
- Stars
- (5,722 stars)
- PR merge metrics
- (PR metrics pending)
Description
@JanuszL Hope all is well on your end.
I have a pipeline that uses Pytorch Lightning + DALI to do pose-estimation. Trained many models successfully by now. We are using Hydra to orchestrate our configs and perform sequential sweeps.
When I run 5-6 sequential jobs using Hydra's multirun -- all goes well.
The problem arises when I run a larger sweep over 20 models, as follows
python scripts/train_hydra.py --multirun training.train_frames=75,100,150,200 \
'model.losses_to_use=[],[pca_multiview],[temporal],[unimodal],[pca_singleview]' \
training.max_epochs=2
Where training.max_epochs=2 first verifies that each of the 20 models above can successfully train for 2 epochs (later i'll drop this argument).
I run into memory issues.
Could not allocate physical storage of size 123535360 on device 0cuda_vm_resource stat dump:
total VM size: 12884901888
currently allocated: 4545120864
peak allocated: 4545120864
allocated_blocks: 69
block size: 67108864
non-freed allocations: 92
total allocations: 132
total deallocations: 40
total unmapping: 0
free pool size: 18446744071829793952Pool map:
================================
VA region 0000040000000000 : 0000040100000000
0000040000000000 In use Mapped 00007F58F4018B30
0000040004000000 In use Mapped 00007F58F4018E90
0000040008000000 In use Mapped 00007F58F0005230
000004000C000000 In use Mapped 00007F58F0005590
0000040010000000 In use Mapped 00007F58F001B040
0000040014000000 In use Mapped 00007F58F03F7240
0000040018000000 In use Mapped 00007F58F0404850
Note that I used the exact same batch_size, sequence_length and file_name for those 5-6 models that trained successfully.
Not sure if my problem is Hydra related? DALI related? something about the multiple models or DALI loaders seems to be affecting memory I think.
Two things to note:
I can report that my lab mate ran a similar sweep overnight without Hydra multirun (on a smaller video), and didn't run into memory issues.
I saw this DALI issue: https://issueexplorer.com/issue/NVIDIA/DALI/3387
LMK what you think, Dan