Open
Description
Hi, I am unable to run CCD, it seems that the Memory usage spikes immediately. I have reduced num_workers
to 0, and also reduced batch size to 8 and also used fp16. But it seems to me that the Dataloader is creating problem only while running train.py
. I was able to run 'test.py` on ARD model.
Metadata
Metadata
Assignees
Labels
No labels