使用PyTorch可视化工具-Tensorboard;
writer.add_graph(model,graph_inputs)#输出模型结构
报错如下
RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same
1 问题描述与分析
使用PyTorch可视化工具-Tensorboard;
writer.add_graph(model,graph_inputs)#输出模型结构
刚开始没有判断,gpu还是cpu模型,
graph_inputs =