train_dataloader = DataLoader(X_train + y_train,
batch_size=BATCH_SIZE,
shuffle=True)
test_dataloader = DataLoader(X_test + y_test,
batch_size=BATCH_SIZE,
shuffle=True)
for _, (data) in enumerate(train_dataloader):
data = Variable(data.view(-1, BATCH_SIZE))
print((data))
sample_data = data.bernoulli()
I don’t know why doesn’t it works but I got error with create sample_data variable. and errors like this.
RuntimeError: Expected p_in >= 0 && p_in <= 1 to be true, but got false. (Could this error message be improved? If so, please report an enhancement request to PyTorch.)
Please help me. thank you.