how to set the MODEL.DEVICE in lazyconfig?

[braindevices]

cfg = model_zoo.get_config("new_baselines/mask_rcnn_R_50_FPN_400ep_LSJ.py", trained=True)
cfg.model.update(input_format='RGB')
cfg.model.roi_heads.box_predictor.update(test_score_thresh=conf_th) # set threshold for this model
model = detectron2.config.instantiate(cfg.model)

The problem is I cannot find the device setting for model, the only device setting I found in example and document is for trainer.

How do I config the model to use cuda or cpu then?

[braindevices]

if I added device="cpu" to the cfg.model, it will throw:

TypeError: GeneralizedRCNN.__init__() got an unexpected keyword argument 'device'