# yaml-language-server: $schema=https://raw.githubusercontent.com/the-database/traiNNer-redux/refs/heads/master/schemas/redux-config.schema.json
####################
# General Settings
####################
name: %scale%_%archname%
scale: %scale%
num_gpu: auto

#####################
# Network Settings
#####################
# Generator model settings
network_g:
  type: %archname%

############################
# Pretrain and Resume Paths
############################
path:
  # Path of the model to convert to ONNX.
  pretrain_network_g: experiments/%scale%_%archname%/models/net_g_ema_1000.safetensors

#########################
# ONNX conversion options
#########################
onnx:
  dynamo: false  # Whether to use the new beta ONNX exporter. Currently not supported on most architectures.
  fp16: true  # Whether to also export to reduced fp16 precision.
  opset: 17  # ONNX opset version, higher is newer.
  use_static_shapes: false  # Whether to convert with static shapes or dynamic shapes. Dynamic shapes are more convenient for supporting a range of input resolutions, but not all architectures support dynamic shapes.
  shape: 1x3x256x256  # The static shape to use in NxCxHxW format, only used if use_static_shapes is true.
  verify: true  # Verify the accuracy of the ONNX model after exporting it.
  optimize: false  # Runs OnnxSlim on the model to potentially slim the model and improve inference speed.