Hi, After not being able to find a persistent enough serialization format I whipped together a small package which translates models into an ONNX graph: [image] GitHub - DrChainsaw/ONNXNaiveNASflux.jl: Import/export ONNX models Import/export ONNX models. Contribute to DrChainsaw/ONNXNa…

Nice work. I see you are still actively developing your package. On the other hand, development of the ONNX.jl package seems abandoned. What’s your take on that? There also an old thread on ONNX here from which it seems that there are no alternatives yet for using ONNX in Flux.

Hi and thanks for showing interest, I also saw that ONNX.jl has seen very few updates but I don’t know why that is. I basically ended up using only the protos from ONNX so the only consequence are some annoying deprecation and overload warnings due to DataFlow (which is deprecated). In other words…

[image] DrChainsaw: I have been thinking about just generating new proto-files using ProtoBuf It may be harder than you think :slight_smile: I ended up generating most of the code and adding TypeProto_Tensor manually (see the result in onnx_pb.jl ). Somehow I missed your post in January and t…

[image] dfdx: It may be harder than you think :slight_smile: Yeah I saw that issue a few weeks ago and it did not do wonders for my motivation to kick off any work with this. I was definitiely thinking to put generic stuff like protos and base OPs in a framework agnostic package (which is no…

[image] DrChainsaw: Maybe tracing mechanism can/should also be made independent and should ideally be alot less code than all the primitives. Would it make sense to use something like Mjolnir here? XLA.jl more or less does this, and ONNX has (reducible) control flow ops IIRC.

Yes, Mjolnir looks like it would be pretty useful for an ONNX exporter. I still have a feeling that making a diffeq solver work in ONNX will be about as pleasant as making it work in powerpoint animations (and maybe as efficient too)…

Do you have any thoughts regarding importing on ONNX models? I’m interested in importing even more than exporting because it will allow to easily use thousands of pretrained models in Julia. ONNX graph should be imported into some graph representation. And if you have such a representation, it might…

@dfdx I guess importing is the somewhat easier part as it translates from a smaller spec to a larger. I don’t have any more thoughts than the obvious that one should be able to pick their framework which in my mind means one package per framework for import as well. One thing that made the import i…

FWIW, I have now registered the package under the name ONNXNaiveNASflux so that breakage can be handled with compat. While the name is probably even more ugly than the previous name, it should make it clear that it is a bit of a niche package when/if more canoncial ONNX packages materializes. I’m s…

[ANN] ONNX export for Flux models

Package Announcements

ToucheSir August 27, 2020, 10:53pm 9

Doesn’t ONNX.jl already do this? The recent discussion in the other thread was that the package seems to have bitrotted and folks are no longer able to import models…

Topic		Replies	Views
Save Flux model to ONNX? General Usage	14	2454	November 3, 2023
Onnx support for Julia ML frameworks Machine Learning question	9	2513	August 24, 2020
How to export a Flux model to Python? Machine Learning flux	2	455	July 19, 2023
Is there a C/C++/Rust/Go library that can do inference on Flux models? Machine Learning question , flux , c	11	533	August 9, 2023
How to import a pre-trained neural network into Flux.jl ? Machine Learning	7	3444	April 24, 2020

[ANN] ONNX export for Flux models

Related topics