From 2d6553850efc4742acbe5a54160b1d6e9ff9d5d9 Mon Sep 17 00:00:00 2001 From: Chun Yang Date: Sun, 5 Nov 2017 17:12:48 -0800 Subject: [PATCH] Change device_id -> device Per https://github.com/pytorch/pytorch/pull/2872 --- rlpytorch/model_interface.py | 2 +- rlpytorch/model_loader.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/rlpytorch/model_interface.py b/rlpytorch/model_interface.py index bcedf172..9a28e52f 100644 --- a/rlpytorch/model_interface.py +++ b/rlpytorch/model_interface.py @@ -96,7 +96,7 @@ def add_model(self, key, model, copy=False, cuda=False, gpu_id=None, opt=False, self.models[key] = model.clone() if copy else model if cuda: if gpu_id is not None: - self.models[key].cuda(device_id=gpu_id) + self.models[key].cuda(device=gpu_id) else: self.models[key].cuda() diff --git a/rlpytorch/model_loader.py b/rlpytorch/model_loader.py index ef471c0e..580eddc7 100644 --- a/rlpytorch/model_loader.py +++ b/rlpytorch/model_loader.py @@ -92,7 +92,7 @@ def load_model(self, params): sys.exit(1) if args.gpu is not None and args.gpu >= 0: - model.cuda(device_id=args.gpu) + model.cuda(device=args.gpu) return model