[训练可视化工具] wandb 小结

Author ：Horizon Max

✨ 编程技巧篇：各种操作小结

神经网络篇：经典网络模型

算法篇：再忙也别忘了 LeetCode

01. wandb 基本操作

官方文档 ：Weights & Biases Docs

️ 初始化

wandb.init(project = "Project_1",
           config = args)

️ 保存数据

for epoch in range(start_epoch, end_epoch):
	wandb.log({'Progress': epoch}, step=epoch)
	
	# save logs
	wandb.log({'Accuracy': train_loss}, step=epoch)
	wandb.log({'Loss': train_loss}, step=epoch)

	# save image
	grid_img = torchvision.utils.make_grid(inputs, normalize=False)
	wandb.log({"Img": wandb.Image(torch.nan_to_num(grid_img.detach().cpu())}, step=epoch)

	# save parameters and gradient
	wandb.watch(net, log='all', log_freq=1)

	wandb.finish()

02. wandb 网页数据不更新，本地程序仍在运行

大多是由于 网络不稳定引起的，等待程序运行结束后，找到本地 wandb 文件，使用以下命令同步到网页 ：
（ 假设文件夹名称为：run-20230316_161158-m7886rzk ）

wandb sync ./run-20230316_161158-m7886rzk