用了append报错,将apend换成extend
loss, I_ZY_bound, I_ZX_bound, I_ZY_bound_2 = model.batch_loss(x_batch, y_batch)
train_loss.append(loss.item())
logits, logits2 = model.forward(x_batch)
用了append报错,将apend换成extend
loss, I_ZY_bound, I_ZX_bound, I_ZY_bound_2 = model.batch_loss(x_batch, y_batch)
train_loss.append(loss.item())
logits, logits2 = model.forward(x_batch)