|
发表于 2022-11-18 20:05:24
|
显示全部楼层
dx12跑第三方模型出现,大佬求解
Error: Input to reshape is a tensor with 8 values, but the requested shape has 50176
[[node gradients/Mean_6_grad/Reshape (defined at D:\DeepFaceLab_DX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\framework\ops.py:1762) ]]
Original stack trace for 'gradients/Mean_6_grad/Reshape':
File "threading.py", line 884, in _bootstrap
File "threading.py", line 916, in _bootstrap_inner
File "threading.py", line 864, in run
File "D:\DeepFaceLab_DX12\_internal\DeepFaceLab\models\ModelBase.py", line 193, in __init__
self.on_initialize()
File "D:\DeepFaceLab_DX12\_internal\DeepFaceLab\models\Model_SAEHD\Model.py", line 547, in on_initialize
gpu_G_loss_gvs += [ nn.gradients ( gpu_G_loss, self.src_dst_trainable_weights )]
File "D:\DeepFaceLab_DX12\_internal\DeepFaceLab\core\leras\ops\__init__.py", line 55, in tf_gradients
grads = gradients.gradients(loss, vars, colocate_gradients_with_ops=True )
File "D:\DeepFaceLab_DX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\ops\gradients_impl.py", line 158, in gradients
unconnected_gradients)
File "D:\DeepFaceLab_DX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\ops\gradients_util.py", line 679, in _GradientsHelper
lambda: grad_fn(op, *out_grads))
File "D:\DeepFaceLab_DX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\ops\gradients_util.py", line 350, in _MaybeCompile
return grad_fn() # Exit early
File "D:\DeepFaceLab_DX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\ops\gradients_util.py", line 679, in <lambda>
lambda: grad_fn(op, *out_grads))
File "D:\DeepFaceLab_DX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\ops\math_grad.py", line 237, in _MeanGrad
sum_grad = _SumGrad(op, grad)[0]
File "D:\DeepFaceLab_DX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\ops\math_grad.py", line 201, in _SumGrad
grad = array_ops.reshape(grad, output_shape_kept_dims)
File "D:\DeepFaceLab_DX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\ops\array_ops.py", line 131, in reshape
result = gen_array_ops.reshape(tensor, shape, name)
File "D:\DeepFaceLab_DX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\ops\gen_array_ops.py", line 9726, in reshape
"Reshape", tensor=tensor, shape=shape, name=name)
File "D:\DeepFaceLab_DX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\framework\op_def_library.py", line 794, in _apply_op_helper
op_def=op_def)
File "D:\DeepFaceLab_DX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\util\deprecation.py", line 507, in new_func
return func(*args, **kwargs)
File "D:\DeepFaceLab_DX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\framework\ops.py", line 3371, in create_op
attrs, op_def, compute_device)
File "D:\DeepFaceLab_DX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\framework\ops.py", line 3440, in _create_op_internal
op_def=op_def)
File "D:\DeepFaceLab_DX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\framework\ops.py", line 1762, in __init__
self._traceback = tf_stack.extract_stack()
...which was originally created as op 'Mean_6', defined at: |
|