Pytorch 1.5.0 (#3973)

* Standard deviation can no longer be set to 0

* Remove torch pinned version

* 9th instead of 10th, silly me
This commit is contained in:
Lysandre Debut
2020-05-05 10:23:01 -04:00
committed by GitHub
parent 818463ee8e
commit 79b1c6966b
2 changed files with 3 additions and 3 deletions

View File

@@ -67,7 +67,7 @@ extras["mecab"] = ["mecab-python3"]
extras["sklearn"] = ["scikit-learn"]
extras["tf"] = ["tensorflow"]
extras["tf-cpu"] = ["tensorflow-cpu"]
extras["torch"] = ["torch==1.4.0"]
extras["torch"] = ["torch"]
extras["serving"] = ["pydantic", "uvicorn", "fastapi", "starlette"]
extras["all"] = extras["serving"] + ["tensorflow", "torch"]