发生异常: RuntimeError CUDA error: the launch timed out and was terminated File "/home/lym/Compare experiment new/classification/main_cifar.py", line 72, in main model = net_module(sparse=args.budget >= 0, pretrained=args.pretrained).to(device=device) File "/home/lym/Compare experiment new/classification/main_cifar.py", line 232, in <module> main()
Opened this issue · 0 comments
liuyiming199721 commented
This code is buggy and the environment configuration is fine, the server is 4090,24 gigabytes of video memory。
model = net_module(sparse=args.budget >= 0, pretrained=args.pretrained).to(device=device)