2023年5月26日
pytorch中tensorboard的使用
摘要： from torch.utils.tensorboard import SummaryWriter writer = SummaryWriter('save') #建立一个保存数据用的东西，save是输出的文件名 dummy_input = torch.rand(512, 1, 28, 28) # 阅读全文
posted @ 2023-05-26 19:49 瘋耔 阅读(16) 评论(0) 推荐(0) 编辑
跳至侧栏