[tune] upgrade gpytorch, bump default pytorch to 1.7.0 (#12776)

* upgrade gpytorch

Signed-off-by: Richard Liaw <rliaw@berkeley.edu>

* pin

Signed-off-by: Richard Liaw <rliaw@berkeley.edu>

* version-torch

Signed-off-by: Richard Liaw <rliaw@berkeley.edu>

* fix-build

Signed-off-by: Richard Liaw <rliaw@berkeley.edu>
This commit is contained in:
Richard Liaw
2020-12-12 10:35:33 -08:00
committed by GitHub
parent 7e09f1d934
commit 2f2bd884a3
4 changed files with 10 additions and 10 deletions
+2 -2
View File
@@ -182,8 +182,8 @@ class GANOperator(TrainingOperator):
@override(TrainingOperator)
def train_batch(self, batch, batch_info):
"""Trains on one batch of data from the data creator."""
real_label = 1
fake_label = 0
real_label = 1.0
fake_label = 0.
discriminator, generator = self.models
optimD, optimG = self.optimizers
-1
View File
@@ -1,5 +1,4 @@
ax-platform
gpytorch==1.2
bayesian-optimization
ConfigSpace==0.4.10
dragonfly-opt