Hi everyone, thanks to Anastasia Golovina you can now use a variety of optimisers (including Adam, AdamW, RMS, Adagrad and SGD) to train TorchScript models from Kotlin - see tests for usage examples.
🙌 1
a
altavir
05/22/2022, 6:00 AM
PR into dev and integration with optimization API would be nice.
р
Ролан
05/22/2022, 6:54 PM
I don't the resources for a cross-platform integration and packaging on the JNI side
a
altavir
05/23/2022, 7:28 AM
You don't have to implement it for all platforms. It is OK to do it only for one platform.