From 420f6f9b16dae38cfe95275bcc4aee5384c1b67e Mon Sep 17 00:00:00 2001 From: Anirudh Dagar Date: Mon, 14 Feb 2022 21:07:33 +0100 Subject: [PATCH] Explicitly move to cpu before plotting (#1966) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Fix TypeError: can’t convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first --- chapter_computer-vision/kaggle-dog.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/chapter_computer-vision/kaggle-dog.md b/chapter_computer-vision/kaggle-dog.md index bd6eb738e5..24ffffca04 100644 --- a/chapter_computer-vision/kaggle-dog.md +++ b/chapter_computer-vision/kaggle-dog.md @@ -416,7 +416,7 @@ def train(net, train_iter, valid_iter, num_epochs, lr, wd, devices, lr_period, measures = f'train loss {metric[0] / metric[1]:.3f}' if valid_iter is not None: valid_loss = evaluate_loss(valid_iter, net, devices) - animator.add(epoch + 1, (None, valid_loss.detach())) + animator.add(epoch + 1, (None, valid_loss.detach().cpu())) scheduler.step() if valid_iter is not None: measures += f', valid loss {valid_loss:.3f}'