From 402bc4afca70d5038d1ccbd63bd95aeef23ee444 Mon Sep 17 00:00:00 2001 From: Younes Belkada Date: Wed, 7 Dec 2022 12:50:02 +0000 Subject: [PATCH] Update README.md --- README.md | 1 + 1 file changed, 1 insertion(+) diff --git a/README.md b/README.md index 31b888a..a8e9c1b 100644 --- a/README.md +++ b/README.md @@ -64,6 +64,7 @@ prediction = torch.nn.functional.interpolate( mode="bicubic", align_corners=False, ) +``` # visualize the prediction output = prediction.squeeze().cpu().numpy()