Zhihu Youtube Twitter Mastodon Rss
HOME
SOFTWARE
DOCS

  • Schola
    Schola

    • AMD Schola documentation

    Home » Manuals » API Documentation » schola.sb3 » schola.sb3.utils

    Manuals home

  • Schola
    Schola

    • schola.sb3.utils

    Utility functions for working with stable baselines 3

    Functions

    convert_ckpt_to_onnx_for_unreal([trainer, …])

    Convert a stable baselines model to ONNX for use in Unreal.

    get_scholasb3_model(model)

    Get the ScholaModel for a stable baselines algorithm.

    save_model_as_onnx(model, export_path)

    Save a stable baselines model as ONNX.

    Classes

    RenderImagesWrapper(venv)

    Renders image observations to an interactive matplotlib window.

    SB3A2CModel(policy, action_space)

    SB3DDPGModel(policy, action_space)

    SB3DQNModel(policy, action_space)

    SB3PPOModel(policy, action_space)

    SB3SACModel(policy, action_space)

    SB3ScholaModel(policy, action_space)

    A PyTorch Module that wraps a stable baselines policy and extracts the necessary components to export to ONNX.

    SB3TD3Model(policy, action_space)

    VecMergeDictActionWrapper(venv)

    A vectorized wrapper for merging a dictionary of actions into 1 single action.

    Related pages

    • Visit the Schola product page for download links and more information.

    Looking for more documentation on GPUOpen?

    AMD GPUOpen software blogs

    Our handy software release blogs will help you make good use of our tools, SDKs, and effects, as well as sharing the latest features with new releases.

    GPUOpen Manuals

    Don’t miss our manual documentation! And if slide decks are what you’re after, you’ll find 100+ of our finest presentations here.

    AMD GPUOpen Performance Guides

    The home of great performance and optimization advice for AMD RDNA™ 2 GPUs, AMD Ryzen™ CPUs, and so much more.

    Getting started: AMD GPUOpen software

    New or fairly new to AMD’s tools, libraries, and effects? This is the best place to get started on GPUOpen!

    AMD GPUOpen Getting Started Development and Performance

    Looking for tips on getting started with developing and/or optimizing your game, whether on AMD hardware or generally? We’ve got you covered!

    AMD GPUOpen Technical blogs

    Browse our technical blogs, and find valuable advice on developing with AMD hardware, ray tracing, Vulkan®, DirectX®, Unreal Engine, and lots more.

    Find out more about our software!

    AMD GPUOpen Effects - AMD FidelityFX technologies

    Create wonder. No black boxes. Meet the AMD FidelityFX SDK!

    AMD GPUOpen Samples

    Browse all our useful samples. Perfect for when you’re needing to get started, want to integrate one of our libraries, and much more.

    AMD GPUOpen developer SDKs

    Discover what our SDK technologies can offer you. Query hardware or software, manage memory, create rendering applications or machine learning, and much more!

    AMD GPUOpen Developer Tools

    Analyze, Optimize, Profile, Benchmark. We provide you with the developer tools you need to make sure your game is the best it can be!