Commit 74670a88 authored by Geoffrey Yu's avatar Geoffrey Yu
Browse files

update input pipeline multimer

parent 324b2ea6
......@@ -138,10 +138,7 @@ def process_tensors_from_config(tensors, common_cfg, mode_cfg):
d["ensemble_index"] = i
return fn(d)
nonensembled = nonensembled_transform_fns(
common_cfg,
mode_cfg,
)
nonensembled = nonensembled_transform_fns()
gt_tensors = compose(grountruth_transforms_fns())(gt_tensors)
tensors = compose(nonensembled)(tensors)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment