The code error in logger

[preesee]

Hi all, I am doing research for SimSiam with this implementation.
The framework of this backbone is really good and easy to expend, so I tried to run the SimSiam on ImageNet and CUB200 datasets.
However, to reproduce the experimental results in the paper, I had to run the model with resnet50/CUB200 with batch size =256, and I got an OOM problem.
Then I ran the model on two GPUs, but the code complains that there are some errors.

Can anyone help me out with this issue?
The code works fine when I only use one GPU
Thanks.
The command I used like:
CUDA_VISICE_DEVICES=0,1 python /root/SimSiam/main.py --data_dir /root/CUB_200_2011/ --log_dir ../logs/ -c configs/simsiam_cub200.yaml --ckpt_dir ~/.cache --hide_progress