wandb를 checkpoint로 설정하면 마지막 파일이 저장되지 않고 에러가 발생한다.
checkpoint 대신 end를 사용했더니 checkpoint로 저장이 되면서 에러도 발생하지 않았다.
# Wandb를 초기화 한다
wandb.init(
entity=config['wandb']['entity'],
project=config['wandb']['project'],
name=config['wandb']['name'])
# checkpoint를 저장하도록 설정한다.
os.environ["WANDB_LOG_MODEL"]="checkpoint" # 이 부분을 end로 수정
wandb.save("./checkpoints/checkpoint-last/*")
wandb.finish()
ckeckpont : 매 epoch 마다 모델을 저장
end : 마지막 epoch의 모델을 저장