def test(train_dict, x, y):
    """ Test network for one iter

    Args:
      train_dict (dict): dictionary containing network, optimizer, loss and loader
      x (torch.tensor): input image from dataset
      y (torch.tensor): actual y value from dataset

    Returns:
      torch.tensor: mse loss
      torch.tensor: point loss
    """

    yhat = train_dict["net"](x)
    mse_loss = train_dict["mse_crit"](yhat, y.float())
    point_loss = train_dict["point_crit"](yhat, y.float())
    return mse_loss, point_loss