def METHOD_NAME(fmodel, params, buffers, sample, target):
    sample = sample.unsqueeze(0)  # prepend batch dimension for processing
    target = target.unsqueeze(0)
    prediction = fmodel(params, buffers, sample)
    return loss_fn(prediction, target)