writer = SummaryWriter('./log') 然后就可以愉快的寫loss到你得這個writer了 niter = epoch * len(train_loader) + i writer.add_scalars(args.result_path + 'Train_val_loss', {args.result_path+'train_loss': loss.data.item()}, niter) 其中,add_scalars是將不同得變量添加到同一個圖下,圖的...
www.dbjr.com.cn/article/1811...htm 2025-6-8