Made torch device configurable
Compare changes
+ 35
− 0
@@ -117,3 +119,36 @@ def instantiate_controller(cfg: DictConfig) -> BaseController:
Made torch device configurable for each device. Provided default implementation so defining the torch device is not mandatory. Defaults to cuda:0 if cuda is available and CPU otherwise.