search

taohan10200 / IIM

PyTorch implementations of the paper: "Learning Independent Instance Maps for Crowd Localization"
MIT License
163 stars 39 forks source link

Memory allocation error #32

Closed akshu281 closed 1 year ago

akshu281 commented 1 year ago

Hello @taohan10200, I tried to use this framework on the custom dataset and I get the below error in trainer.py

crop_imgs, crop_gt, crop_masks = map(lambda x: torch.cat(x, dim=0), (crop_imgs, crop_gt, crop_masks))

image

Any workaround to solve this? I use a 40GB RAM to do this training

aviendre commented 8 months ago

Hi, how did you manage to solve this problem as I am running on 59 GB RAM