Just run python main.py, I have the following issue: "z_grouped = torch.cat(test_z, dim=0).view(-1, c.n_transforms_test, c.n_feat) RuntimeError: shape '[-1, 64, 768]' is invalid for input of size 262144" could you help to check?
Just run python main.py, I have the following issue:
"z_grouped = torch.cat(test_z, dim=0).view(-1, c.n_transforms_test, c.n_feat)
RuntimeError: shape '[-1, 64, 768]' is invalid for input of size 262144"
could you help to check?