all AI news
Convert Pegasus model to ONNX [Discussion]
Sept. 22, 2022, 9:17 p.m. | /u/Lost-Letterhead2105
Machine Learning www.reddit.com
I am working on a project where I fine-tuned a Pegasus model on the Reddit dataset. Now, I need to convert the [fine-tuned model](https://huggingface.co/QuickRead/pegasus-reddit-7e05-new) to ONNX for the deployment stage. I have followed this guide from Huggingface to [convert to the ONNX model for unsupported architect](https://huggingface.co/docs/transformers/serialization#implementing-a-custom-onnx-configuration)s. I got it done but the ONNX model can't generate text. Turned out that Pegasus is an encoder-decoder model and most guides are for either encoder-model (e.g. BERT) or decoder-model (e.g. GPT2). …
More from www.reddit.com / Machine Learning
Jobs in AI, ML, Big Data
Data Scientist (m/f/x/d)
@ Symanto Research GmbH & Co. KG | Spain, Germany
Automated Greenhouse Expert - Phenotyping & Data Analysis (all genders)
@ Bayer | Frankfurt a.M., Hessen, DE
Machine Learning Scientist II
@ Expedia Group | India - Bengaluru
Data Engineer/Senior Data Engineer, Bioinformatics
@ Flagship Pioneering, Inc. | Cambridge, MA USA
Intern (AI lab)
@ UL Solutions | Dublin, Co. Dublin, Ireland
Senior Operations Research Analyst / Predictive Modeler
@ LinQuest | Colorado Springs, Colorado, United States