解决方案：改成离线模式

import os
import wandb
os.environ[“WANDB_API_KEY”] = ‘KEY’
os.environ[“WANDB_MODE”] = “offline”

原因：
使用wandb在线模式运行代码，服务器是一边运行我们的代码一边向wandb官网上传我们的数据，导致服务器显卡占用比特别大