Traceback (most recent call last):] File "run.py", line 295, in <module> trainer.train() File "run.py", line 88, in train assert self.norm != 0 AssertionError