Function preprocess_batch

evaluate.py:23–27

(batch)

        av_dict[key] = [value]

def preprocess_batch(batch):
    for k, v in batch.items():
        if isinstance(v, Tensor):
            batch[k] = v.unsqueeze(0).cuda()
    return batch

if __name__ == "__main__":
    parser = get_eval_parser()