iceandcoal

The above exception was the direct cause of the following exception: Traceback (most recent call last): File "/root/autodl-tmp/fo/fakeout/avhubert/xunlian.py", line 429, in app.run(main) File "/root/miniconda3/envs/df/lib/python3.8/site-packages/absl/app.py", line 312, in run _run_main(main, args)...

Hello author, can you give the code of the training model to save the weight?

NameError: unbound axis name: i. The following axis names (e.g. defined by pmap) are available to collective operations: [] Hello，How to solve this problem?