Edward Fan

I got the same problem. And if I change the config file to vit_pipeline.py, the error will be : ```python TypeError: layer_norm(): argument 'input' (position 1) must be Tensor, not...

Thanks, Liu. I pulled the latest codes of ColossalAI and ColossalAi-Examples, then I got another error about `titans`: ```python Traceback (most recent call last): File "train_with_cifar10.py", line 13, in from...

Thanks, Liu. The problem was solved by reinstalling `titans`. But the training process will be stuck at the `86/196` step. I used 4 A6000 GPUs with `colossalai run --nproc_per_node 4...