Samsung / ONE

On-device Neural Engine
Other
430 stars 157 forks source link

[onert/train] Let's introduce circle+ generating tools #12650

Closed zetwhite closed 4 months ago

zetwhite commented 7 months ago

There is no way to create a circle+ file in public. So, Let's generate tools for generating circle+, More exactly tools that inject training parameters into the circle file.

zetwhite commented 7 months ago

about draft

I made a draft (https://github.com/Samsung/ONE/pull/12649) that injects training parameters into circle, referencing circle/oxri(inner source) implmentation.

I need more ideas :bulb:

After I made this draft work somehow, I started to think that it is NOT helpful because of the input format.

[ python API ]

IMHO, python API is more useful than command line executing tools. But most of the tools in ONE are kinds of command line tools, so I'm hesitant to work on it.

#simple pseudo API 
import onert # new package

training_pram = onert.TrainingParameter(SGD(learning_rate = 0.01), MSE(from_logits=true), batch_size=32)

circle = onert.open("mnist.circle")
circle.inject_tparam(training_param)
circle.export("mnist_with_meta.circle")

/cc @Samsung/one_onert

zetwhite commented 7 months ago

After some talks with @Samsung/one_onert yesterday, I rethink this issue.

zetwhite commented 4 months ago

basic features are all added in https://github.com/Samsung/ONE/tree/master/tools/circle_plus_gen. So close this issue