validation

validation.py

model.eval() # 取消dropout or batch normalization
with torch.no_grad(): # 不計算梯度
  predict = []
  for data, labels in test_loader:
    # Forward
    out = model(data)
    predict.append(out.numpy())
model.train() # 重新開啟dropout or batch normalization