Option to normalize latent interpolation images (Lightning-Universe#438)

* add option to normalize latent interpolation images * linspace * update Co-authored-by: ananyahjha93 <[email protected]>
chris-clem · Dec 17, 2020 · ce4db83 · ce4db83
1 parent 141a7d3
commit ce4db83
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pl_bolts/callbacks/variational.py b/pl_bolts/callbacks/variational.py
@@ -1,4 +1,5 @@
 import math
+import numpy as np
 
 import numpy as np
 import torch