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.