(cfg: FlowPredictorCfg)
| 15 | |
| 16 | |
| 17 | def get_flow_predictor(cfg: FlowPredictorCfg) -> FlowPredictor: |
| 18 | flow_predictor = FLOW_PREDICTORS[cfg.name](cfg) |
| 19 | convert_to_buffer(flow_predictor, persistent=False) |
| 20 | return flow_predictor |
| 21 | |
| 22 | |
| 23 | @torch.no_grad() |
no test coverage detected