OpenGVLab / UniFormerV2

[ICCV2023] UniFormerV2: Spatiotemporal Learning by Arming Image ViTs with Video UniFormer
https://arxiv.org/abs/2211.09552
Apache License 2.0
294 stars 19 forks source link

RuntimeError: unable to mmap 276 bytes from file </torch_116128_1191909093_62920>: Cannot allocate memory (12) #78

Open ahmadjaved97 opened 2 months ago

ahmadjaved97 commented 2 months ago

While trying to run the training, whenever I put the NUM_GPU > 1, it shows the following error Screenshot from 2024-09-26 00-44-31 . I am using the following slurm command to allocate the resources srun --pty --nodes=1 --ntasks-per-node=8 --cpus-per-task=8 --time=7-00:00:00 --partition=gpu_partition --gpus=4 /bin/bash. I am trying to train a multilabel dataset