iree-org / iree-torch

Torch Frontend for IREE
Apache License 2.0
25 stars 11 forks source link

Remove dependency on sklearn for regression example. #69

Closed dellis23 closed 1 year ago

dellis23 commented 1 year ago

And numpy.

silvasean commented 1 year ago

Also in the example in the IREE repo it would be good to generate a larger amount of random data on the fly there -- there isn't really that much signal that the loss is going down with a tiny number of examples.