Pytorch를 사용하고 데이터를 tensorboardX에 기록하고 체크 포인트를 저장하는 강화 학습 훈련 스크립트를 실행했습니다. 이제 계속 훈련하고 싶습니다. 중단 한 부분부터 계속하도록 tensorboardX에 어떻게 지시합니까? 감사합니다!
훈련 플롯을 계속하는 방법을 알아 냈습니다. 요약 작성기를 만드는 log_dir
동안 처음 교육 할 때 사용한 것과 동일한 내용을 제공해야합니다 .
from tensorboardX import SummaryWriter
writer = SummaryWriter('log_dir')
그런 다음 학습 루프 내에서 단계가 0이 아닌 왼쪽부터 시작해야합니다.
writer.add_scalar('average reward',rewards.mean(),step)
이 기사는 인터넷에서 수집됩니다. 재 인쇄 할 때 출처를 알려주십시오.
침해가 발생한 경우 연락 주시기 바랍니다[email protected] 삭제
몇 마디 만하겠습니다