Function inference

DeepfakeBench_DF40/training/test.py:142–144

(model, data_dict)

@torch.no_grad()
def inference(model, data_dict):
    predictions = model(data_dict, inference=True)
    return predictions


def main():