runtime error

_impl return forward_call(*args, **kwargs) File "/home/user/app/stylegan_human/training_scripts/sg2/training/networks.py", line 498, in forward x, img = block(x, img, cur_ws, **block_kwargs) File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl return forward_call(*args, **kwargs) File "/home/user/app/stylegan_human/training_scripts/sg2/training/networks.py", line 426, in forward x = self.conv0(x, next(w_iter), fused_modconv=fused_modconv, **layer_kwargs) File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl return forward_call(*args, **kwargs) File "/home/user/app/stylegan_human/training_scripts/sg2/training/networks.py", line 312, in forward x = modulated_conv2d(x=x, weight=self.weight, styles=styles, noise=noise, up=self.up, File "/home/user/app/torch_utils/misc.py", line 103, in decorator return fn(*args, **kwargs) File "/home/user/app/stylegan_human/training_scripts/sg2/training/networks.py", line 67, in modulated_conv2d x = conv2d_resample.conv2d_resample(x=x, w=weight.to(x.dtype), f=resample_filter, up=up, down=down, padding=padding, flip_weight=flip_weight) File "/home/user/app/torch_utils/misc.py", line 103, in decorator return fn(*args, **kwargs) File "/home/user/app/torch_utils/ops/conv2d_resample.py", line 125, in conv2d_resample x = _conv2d_wrapper(x=x, w=w, stride=up, padding=[pyt,pxt], groups=groups, transpose=True, flip_weight=(not flip_weight)) File "/home/user/app/torch_utils/ops/conv2d_resample.py", line 41, in _conv2d_wrapper return op(x, w, stride=stride, padding=padding, groups=groups) File "/home/user/app/torch_utils/ops/conv2d_gradfix.py", line 43, in conv_transpose2d return torch.nn.functional.conv_transpose2d(input=input, weight=weight, bias=bias, stride=stride, padding=padding, output_padding=output_padding, groups=groups, dilation=dilation) RuntimeError: "slow_conv_transpose2d_out_cpu" not implemented for 'Half'

Container logs: