parsnip.ParsnipModel.to

ParsnipModel.to(device, force=False)

Send the model to the specified device

Parameters:
  • device (str) – PyTorch device

  • force (bool, optional) – If True, force the model to be sent to the device even if it is there already (useful if only parts of the model are there), by default False