Parametrized types #49
Description
At the moment, all of the models (well, actually everything) use the torch.double data-type by default, but it would be nice to have this be an optional argument to the model constructors.
{{ message }}
At the moment, all of the models (well, actually everything) use the torch.double data-type by default, but it would be nice to have this be an optional argument to the model constructors.