Function get_cost

ann_class2/pytorch_example2.py:89–97

(model, loss, inputs, labels)

# similar to train() but not doing the backprop step
def get_cost(model, loss, inputs, labels):
  inputs = Variable(inputs, requires_grad=False)
  labels = Variable(labels, requires_grad=False)

  # Forward
  logits = model.forward(inputs)
  output = loss.forward(logits, labels)

  return output.item()


# define the prediction procedure