Hi, I troubling with train tensors. here is the code…
X_train, X_test, y_train, y_test = train_test_split(
(db1_sig + db2_sig),
(db1_label + db2_label) * 2,
test_size=0.33,
shuffle=True
)
print()
train_dataloader = DataLoader(X_train,
batch_size=BATCH_SIZE,
num_workers=0, collate_fn=lambda x: x,
shuffle=True)
test_dataloader = DataLoader(X_test,
batch_size=BATCH_SIZE,
shuffle=True)
...
loss_ = []
for epoch in range(EPOCH):
'''First bbrbm'''
for _, (data) in enumerate(train_dataloader):
print(len(data[0]))
data = torch.tensor(data, dtype=torch.float32)
and error contents are like this.
ValueError: expected sequence of length 1300000 at dim 1 (got 650000)
I think data variable is a problem because data is list that made of 650000 length numpy arrays. I still figuring out what is wrong with this code and plz help anyone. thanks