Function build_flowformer

core/FlowFormer/__init__.py:2–9

(cfg)

import torch
def build_flowformer(cfg):
    name = cfg.transformer 
    if name == 'latentcostformer':
        from .LatentCostFormer.transformer import FlowFormer
    else:
        raise ValueError(f"FlowFormer = {name} is not a valid architecture!")

    return FlowFormer(cfg[name])