add two dev packages for training

5 years ago · 3da368247a
parent 805bf29206
commit 3da368247a
2 changed files with 548 additions and 431 deletions
--- a/2
+++ b/2
@ -76,6 +76,8 @@ parameterized = "*"
 torch = "*"
 torchsummary = "*"
 blosc = "==1.9.2"
+albumentations = "*"
+segmentation-models-pytorch = "*"

 [packages]
 atomicwrites = "*"
--- a/Pipfile.lock
+++ b/Pipfile.lock