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)
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)