Fine-tune neural graphics models using Model Gym
Introduction
Install Model Gym and explore neural graphics examples
Set up your environment
Launch the training notebook
Fine-tune and export a quantized model
Inspect the model graph with Model Explorer
Next Steps

  Log an issue

  Fork and edit

  Discuss on Discord

About this Learning Path

Skill level:  Advanced
Reading time:  45 min
Last updated:  18 Aug 2026
Skill level:
  Advanced
Reading time:
  45 min
Last updated:
  18 Aug 2026
Author:Annie Tallund, Arm
Arm IP:Mali
Tags: ML Linux PyTorch Jupyter Notebook Vulkan NX
Author:
  • Annie Tallund, Arm
Arm IP:
Mali
Tags:
ML Linux PyTorch Jupyter Notebook Vulkan NX

Who is this for?

This is an advanced topic for developers exploring neural graphics and interested in training and deploying upscaling models like Neural Super Sampling (NSS) using PyTorch and Arm’s hardware-aware backend.

What will you learn?

Upon completion of this Learning Path, you will be able to:

  • Understand the principles of neural graphics and how it’s applied to game performance
  • Learn how to fine-tune and evaluate a neural network for Neural Super Sampling (NSS)
  • Use the Model Gym Python API and CLI to configure and train neural graphics models
  • Fine-tune an NSS model with quantization-aware training (QAT) and export it to .vgf
  • Inspect the graph of exported .vgf models using Model Explorer

Prerequisites

Before starting, you will need the following:

  • Basic understanding of PyTorch and machine learning concepts
  • A development machine running Ubuntu 22.04 or later, with a CUDA-capable NVIDIA® GPU
  • CUDA Toolkit v13.1.1 or later

Summary

AI-assisted

This summary was drafted with an approved AI-assisted workflow and reviewed by Arm contributors before publication. Human technical review remains part of the process so the final page reflects engineering rigor, accuracy, and Arm editorial standards.

Close
?
You’ll use Model Gym notebooks on Ubuntu to train and evaluate a Neural Super Sampling model in PyTorch. Inspect checkpoints, choose post-training or quantization-aware training, and prepare an INT8 export. Then export the fine-tuned model as a .vgf package and validate it in Model Explorer with the VGF adapter.

Frequently asked questions

AI-assisted

These FAQs were drafted with an approved AI-assisted workflow and reviewed by Arm contributors before publication. Human technical review remains part of the process so the final page reflects engineering rigor, accuracy, and Arm editorial standards.

Close
?
What should I verify before creating the Python environment?
Check that python3 reports a supported version (3.10, 3.11, or 3.12). Then install the listed system packages before setting up the examples.
Which notebook should I use to export a deployable model?
Use the model_qat_example.ipynb notebook. It is the only example notebook that includes the export step to produce a .vgf file.
When should I choose PTQ versus QAT for NSS?
Post-training quantization (PTQ) is faster to try because it calibrates an already-trained model. Quantization-aware training (QAT) simulates lower precision during fine-tuning to help preserve accuracy when exporting an INT8 model.
How do I confirm that the export completed correctly?
Expect a .vgf model produced by the export pipeline. Open it in Model Explorer with the VGF adapter; a valid graph with layers, tensor shapes, and connectivity indicates a successful export.
Can Model Explorer show image quality or frame comparisons?
No. Model Explorer visualizes the network structure and execution graph, not rendered output quality. Use the evaluation notebook to compare model output with the ground truth.
Next