jpsember / java-ml

Java classes for machine learning
0 stars 0 forks source link

Add batch normalization option #58

Closed jpsember closed 2 years ago

jpsember commented 2 years ago

Relevant:

https://androidkt.com/use-the-batchnorm-layer-in-pytorch/