for-ai / iterative-data-selection

19 stars 5 forks source link

复现问题 #7

Open chinesezkh opened 1 week ago

chinesezkh commented 1 week ago

作者你好: 非常感谢有如此令人惊叹的成果,我想问一下: torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 44.00 MiB (GPU 0; 23.54 GiB total capacity; 21.31 GiB already allocated; 67.25 MiB free; 22.04 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF 我的设备是一台4090,在这个设备上我可以复现论文吗?

chinesezkh commented 1 week ago

image 如何修改参数,可以在A6000或者4090上复现?