使用的设备是4060 LAPTOP
当我尝试可视化训练的AGW模型时候出现下述报错,显存足够但无法调用(可以完成训练,以及训练中的验证)
torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 1.55 GiB (GPU 0; 8.00 GiB total capacity; 335.96 MiB already allocated; 5.59 GiB free; 1.32 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
使用的设备是4060 LAPTOP 当我尝试可视化训练的AGW模型时候出现下述报错,显存足够但无法调用(可以完成训练,以及训练中的验证)
torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 1.55 GiB (GPU 0; 8.00 GiB total capacity; 335.96 MiB already allocated; 5.59 GiB free; 1.32 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
我的运行脚本如下
python demo/visualize_result.py --config-file logs/market1501/agw_R50/config.yaml --vis-label --dataset-name Market1501 --output logs/agw_R50_Market1501_vis --opts MODEL.WEIGHTS logs/market1501/agw_R50/model_best.pth
我的config文件如下