May I ask why Python main.py -- train False appears after running it X_ Train=torch Tensor (X_train). permute (0,3,1,2) RuntimeError: number of dims don't match in permute

[love-coast]

May I ask why Python main.py -- train False appears after running it X_ Train=torch Tensor (X_train). permute (0,3,1,2) RuntimeError: number of dims don't match in permute This is the result of the operation： D:\STSTNet-master> python main.py --train False lr=0.000050, epochs=800, device=cpu

Subject: 006 Traceback (most recent call last): File "main.py", line 233, in main(config) File "main.py", line 153, in main X_train = torch.Tensor(X_train).permute(0,3,1,2) RuntimeError: number of dims don't match in permute