To train on GPU, use the gpu param, which can be an integer (which gives the index of the gpu) and can be True (will use the first gpu). For example, CalculatorTorch(model, gpu=0) and CalculatorTorch(model, gpu=True).
Same for multiple species calculator CalculatorTorchSeparateSpecies.
To train on GPU, use the
gpu
param, which can be an integer (which gives the index of the gpu) and can beTrue
(will use the first gpu). For example,CalculatorTorch(model, gpu=0)
andCalculatorTorch(model, gpu=True)
.Same for multiple species calculator
CalculatorTorchSeparateSpecies
.