File size: 411 Bytes
2a41a22 |
1 2 3 4 5 6 7 8 9 10 |
from inference import inference
from evaluation.evaluate import evaluate
def valid(model, data_loader_test, pred_dir, method='tmp_val', testset='DIS-VD', only_S_MAE=True, device=0):
model.eval()
inference(model, data_loader_test, pred_dir, method, testset, device=device)
performance_dict = evaluate(pred_dir, method, testset, only_S_MAE=only_S_MAE, epoch=model.epoch)
return performance_dict
|