Add separate setters for autocast

mlverse / torch

R Interface to Torch

https://torch.mlverse.org

Other

490 stars 64 forks source link

Closed dfalbel closed 7 months ago

dfalbel commented 7 months ago

This allows for more fine control of the autocast functionality. Required for https://github.com/mlverse/luz/pull/141