missing 1 required argument "num_spatial_dims"

[wanghao-cst]

Hi, when I run the first command to train the supermobile model, threre is an error as shown in the pict. Maybe it is caused by the wrong torch version, my version is 1.12, could u ple have a check?

[jxhnx]

Try replacing this line by:
output_padding = self._output_padding(x, output_size, self.stride, self.padding, self.kernel_size, num_spatial_dims=2)

It's due to a PyTorch update, at some point the argument wasn't required and 2D the default as far as I remember.

[wanghao-cst]

Thx, that works. may I know your torch version?

[jxhnx]

I'm using some nightlies, 1.13.x 1.14.x, that's why I had to adjust that, too. I never trained the supernet though, so please apply that with caution. With subnets, all works as expected on my end.