|
发表于 2021-10-12 11:45:07
|
显示全部楼层
本帖最后由 slib 于 2021-10-12 11:53 编辑
D:\GPEN-GFPGAN-MSOLD\_internal\Python37\lib\site-packages\torch\nn\functional.py:718: UserWarning: Named tensors and all their associated APIs are an experimental feature and subject to change. Please do not use them for anything important until they are released as stable. (Triggered internally at ..\c10/core/TensorImpl.h:1156.)
return torch.max_pool2d(input, kernel_size, stride, padding, dilation, ceil_mode)
Traceback (most recent call last):
File "face_enhancement_BFR512.py", line 101, in <module>
img, orig_faces, enhanced_faces = faceenhancer.process(im)
File "face_enhancement_BFR512.py", line 43, in process
facebs, landms = self.facedetector.detect(img)
File "D:\GPEN-GFPGAN-MSOLD\GPEN\retinaface\retinaface_detection.py", line 67, in detect
loc, conf, landms = self.net(img) # forward pass
File "D:\GPEN-GFPGAN-MSOLD\_internal\Python37\lib\site-packages\torch\nn\modules\module.py", line 1051, in _call_impl
return forward_call(*input, **kwargs)
File "D:\GPEN-GFPGAN-MSOLD\GPEN\retinaface\facemodels\retinaface.py", line 108, in forward
out = self.body(inputs)
File "D:\GPEN-GFPGAN-MSOLD\_internal\Python37\lib\site-packages\torch\nn\modules\module.py", line 1051, in _call_impl
return forward_call(*input, **kwargs)
File "D:\GPEN-GFPGAN-MSOLD\_internal\Python37\lib\site-packages\torchvision\models\_utils.py", line 62, in forward
x = module(x)
File "D:\GPEN-GFPGAN-MSOLD\_internal\Python37\lib\site-packages\torch\nn\modules\module.py", line 1051, in _call_impl
return forward_call(*input, **kwargs)
File "D:\GPEN-GFPGAN-MSOLD\_internal\Python37\lib\site-packages\torch\nn\modules\container.py", line 139, in forward
input = module(input)
File "D:\GPEN-GFPGAN-MSOLD\_internal\Python37\lib\site-packages\torch\nn\modules\module.py", line 1051, in _call_impl
return forward_call(*input, **kwargs)
File "D:\GPEN-GFPGAN-MSOLD\_internal\Python37\lib\site-packages\torchvision\models\resnet.py", line 136, in forward
identity = self.downsample(x)
File "D:\GPEN-GFPGAN-MSOLD\_internal\Python37\lib\site-packages\torch\nn\modules\module.py", line 1051, in _call_impl
return forward_call(*input, **kwargs)
File "D:\GPEN-GFPGAN-MSOLD\_internal\Python37\lib\site-packages\torch\nn\modules\container.py", line 139, in forward
input = module(input)
File "D:\GPEN-GFPGAN-MSOLD\_internal\Python37\lib\site-packages\torch\nn\modules\module.py", line 1051, in _call_impl
return forward_call(*input, **kwargs)
File "D:\GPEN-GFPGAN-MSOLD\_internal\Python37\lib\site-packages\torch\nn\modules\batchnorm.py", line 178, in forward
self.eps,
File "D:\GPEN-GFPGAN-MSOLD\_internal\Python37\lib\site-packages\torch\nn\functional.py", line 2282, in batch_norm
input, weight, bias, running_mean, running_var, training, momentum, eps, torch.backends.cudnn.enabled
RuntimeError: CUDA out of memory. Tried to allocate 636.00 MiB (GPU 0; 4.00 GiB total capacity; 1.88 GiB already allocated; 282.65 MiB free; 2.52 GiB reserved in total by PyTorch)
请按任意键继续. . . 我开着任务管理器看gpu使用情况,根本没使用 然后使用新的那个gui界面cpu使用率也没上过50%,内存占用率没变化稳定在50,处理进度不动,清理过缓存了
|
|