From b5843ae2be231d66ecfb6f4aeeacac7cefc395ee Mon Sep 17 00:00:00 2001 From: Preyas Garg Date: Tue, 18 Jan 2022 17:41:57 +0530 Subject: [PATCH] Update 16_tensorboard.py model was on graphic card and example_data was on cpu so was throwing error --- 16_tensorboard.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/16_tensorboard.py b/16_tensorboard.py index 0bd67dc6..a1cbd9a8 100644 --- a/16_tensorboard.py +++ b/16_tensorboard.py @@ -80,7 +80,7 @@ def forward(self, x): optimizer = torch.optim.Adam(model.parameters(), lr=learning_rate) ############## TENSORBOARD ######################## -writer.add_graph(model, example_data.reshape(-1, 28*28)) +writer.add_graph(model, example_data.reshape(-1, 28*28).to(device)) #writer.close() #sys.exit() ################################################### @@ -156,4 +156,4 @@ def forward(self, x): preds_i = class_preds[:, i] writer.add_pr_curve(str(i), labels_i, preds_i, global_step=0) writer.close() - ################################################### \ No newline at end of file + ###################################################