Remove torch.set_default_dtype() from package

cics-nd / gptorch

Gaussian processes with PyTorch

MIT License

30 stars 8 forks source link

Remove torch.set_default_dtype() from package #14

Closed sdatkinson closed 5 years ago

sdatkinson commented 5 years ago

torch.set_default_dtype() seems to have a global effect. When used internally in modules, this causes surprising behaviors when using gptorch alongside other PyTorch code that uses other dtypes.

This PR removes those instances in simply uses util.TensorType instead to prevent external effects.