Add einops to requirements

f-dangel · Dec 8, 2020 · f2a5b52 · f2a5b52
1 parent c2478e0
commit f2a5b52
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,2 +1,3 @@
 torch >= 1.6.0, < 2.0.0
 torchvision >= 0.7.0, < 1.0.0
+einops >= 0.3.0