Introduction

This repository implements the Local Dynamics Model in PyTorch.

Code Structure

The structure of this project is based on this MuZero implementation.

Domains

• Block Stacking 3:
• House Building 2:
• Bottle Arrangement:
• Bin Packing:

Getting Started

Installation

Train

python train block_stacking_3 --results_path=model_name

Test

python test block_stacking_3 model_name

Config

Model Architectures

Local Dynamics Model

Q-Value Model

State-Value Model