in flows.py [0:0]
def setup(self):
self.transforms = []
for i in range(self.n_transforms):
mod = hydra.utils.instantiate(
self.single_transform_cfg,
manifold=self.manifold
)
self.transforms.append(mod)
# hack for https://github.com/google/flax/issues/524
key = f'transform{i:02d}'
setattr(self, key, mod)