alibi icon indicating copy to clipboard operation
alibi copied to clipboard

Device management in CFRL Explainers

Open mauicv opened this issue 2 years ago • 0 comments

The counterfactual reinforcement learning explainers don't currently allow the user to specify the device that alibi uses to train the actor and critic models on. Instead, the behaviour is to default to using cuda if available. If the user trains the autoencoder on CPU will this cause errors?

mauicv avatar Feb 09 '22 12:02 mauicv