Running the code model_flops = flops.total () will result in an assertion error？

[liaochuanlin]

Traceback (most recent call last):
File "../unetr_pp/run/run_training.py", line 171, in
main()
File "../unetr_pp/run/run_training.py", line 142, in main
trainer.initialize(not validation_only)
File "/home/lcl/unetr_plus_plus/unetr_pp/training/network_training/unetr_pp_trainer_tumor.py", line 150, in initialize
self.initialize_network()
File "/home/lcl/unetr_plus_plus/unetr_pp/training/network_training/unetr_pp_trainer_tumor.py", line 189, in initialize_network
model_flops = flops.total()
File "/home/lcl/anaconda3/envs/unetr_pp/lib/python3.8/site-packages/fvcore/nn/jit_analysis.py", line 248, in total
stats = self._analyze()
File "/home/lcl/anaconda3/envs/unetr_pp/lib/python3.8/site-packages/fvcore/nn/jit_analysis.py", line 583, in _analyze
op_counts = self._op_handles[kind](inputs, outputs)
File "/home/lcl/anaconda3/envs/unetr_pp/lib/python3.8/site-packages/fvcore/nn/jit_handles.py", line 181, in einsum_flop_jit
assert len(inputs) == 2, len(inputs)

[Amshaker]

Duplicate of #55