search

lettucecfd / lettuce

Computational Fluid Dynamics based on PyTorch and the Lattice Boltzmann Method
MIT License
210 stars 39 forks source link

[Feature]: Context should also work with string input as device #223

Closed PhiSpel closed 1 month ago

PhiSpel commented 1 month ago

Is your proposal related to a problem?

Passing a torch.Device to Context is tedious, when it can already just be a string (it is interpreted correctly).

Describe the solution you'd like

Just change the typing in Context