trying pytorch
@@ -27,7 +27,7 @@ requirements:
- python
- scipy
- simpleitk <2
- - torch >=1.1
+ - pytorch >=1.1
- tqdm
run:
- click
@@ -38,7 +38,7 @@ requirements:
- python
- scipy
- simpleitk <2
- - torch >=1.1
+ - pytorch >=1.1
- tqdm
test:
