...

2021-02-02 23:50:25 -07:00 · 2021-02-02 23:50:25 -07:00 · d7bec392dd
commit d7bec392dd
parent b0a8fa00bc
1 changed files with 1 additions and 0 deletions
--- 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, \