error caused by updated FrEIA

[WeianMao]

Traceback (most recent call last):
File "/media/data/weian/project/imagenet/conditional_INNs/mnist_cINN/train.py", line 28, in
import model
File "/media/data/weian/project/imagenet/conditional_INNs/mnist_cINN/model.py", line 42, in
nodes.append(Node([nodes[-1].out0], flattening_layer, {}, name='flatten'))
NameError: name 'flattening_layer' is not defined

[Brittany-Chen]

Hi, I meet the same problem, have you solved it?

[yuffon]

@WeianMao @Brittany-Chen Have you solved this problem?