Objective.update() allows passing tensors not consistent with self.device. If update() is used to update all the variables in the objective, NLLS optimizers will generate delta on the objective's original device, and then get a runtime error when trying to retract variables that are in a different device.
Objective.update()
allows passing tensors not consistent withself.device
. Ifupdate()
is used to update all the variables in the objective, NLLS optimizers will generatedelta
on the objective's original device, and then get a runtime error when trying to retract variables that are in a different device.