Yes, I change it to:
writer = SummaryWriter(check_point_dir, flush_secs=1, write_to_disk=True, max_queue=1)
Still doesn’t work. When I terminate the training or train it on cpu, the summary can be written to disk. So wired.