From d7bec392ddf919e0114adc44e46f4ad93cba0866 Mon Sep 17 00:00:00 2001
From: James Betker <jbetker@gmail.com>
Date: Tue, 2 Feb 2021 23:50:25 -0700
Subject: [PATCH] ...

---
 codes/models/vqvae/vqvae_3_hardswitch.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/codes/models/vqvae/vqvae_3_hardswitch.py b/codes/models/vqvae/vqvae_3_hardswitch.py
index 9374b432..1222a5ba 100644
--- a/codes/models/vqvae/vqvae_3_hardswitch.py
+++ b/codes/models/vqvae/vqvae_3_hardswitch.py
@@ -1,4 +1,5 @@
 import torch
+import torchvision
 from torch import nn
 
 from models.switched_conv.switched_conv_hard_routing import SwitchedConvHardRouting, \