Traceback (most recent call last): File "/Users/henry/PycharmProjects/algorithm/DSSMv/demo_dssm.py", line 186, in <module> model.fit(train_model_input, train[target].values, batch_size=256, epochs=10, verbose=2, validation_split=0.2) File "/Users/henry/PycharmProjects/algorithm/DSSMv/model/base_tower.py", line 236, in fit train_tensor_data = Data.TensorDataset(torch.from_numpy( File "/Users/henry/miniforge3/envs/py38/lib/python3.8/site-packages/torch/utils/data/dataset.py", line 365, in __init__ assert all(tensors[0].size(0) == tensor.size(0) for tensor in tensors), "Size mismatch between tensors" AssertionError: Size mismatch between tensors
在 base_tower 代码里：
train_tensor_data = Data.TensorDataset(torch.from_numpy( np.concatenate(x, axis=-1)), torch.from_numpy(np.array(y)))
报错了
Traceback (most recent call last): File "/Users/henry/PycharmProjects/algorithm/DSSMv/demo_dssm.py", line 186, in <module> model.fit(train_model_input, train[target].values, batch_size=256, epochs=10, verbose=2, validation_split=0.2) File "/Users/henry/PycharmProjects/algorithm/DSSMv/model/base_tower.py", line 236, in fit train_tensor_data = Data.TensorDataset(torch.from_numpy( File "/Users/henry/miniforge3/envs/py38/lib/python3.8/site-packages/torch/utils/data/dataset.py", line 365, in __init__ assert all(tensors[0].size(0) == tensor.size(0) for tensor in tensors), "Size mismatch between tensors" AssertionError: Size mismatch between tensors在 base_tower 代码里：
train_tensor_data = Data.TensorDataset(torch.from_numpy( np.concatenate(x, axis=-1)), torch.from_numpy(np.array(y)))报错了