self.devices = [ torch.device("cuda:{}".format(i)) for i, ... self.model_gpu_towers.append(model_copy.to(self.devices[i])) if hasattr(self, "target_model"): ...
確定! 回上一頁