These models are part of the Traffic Management System 2 (TMS2) project, an intelligent traffic control system using deep learning and reinforcement learning.

Training Details

Framework: PyTorch
Training Platform: Google Colab (T4 GPU)
Training Date: December 2025

Usage

python

import torch

# Load model
model = torch.load('model.pt')
model.eval()

# Inference
with torch.no_grad():
    output = model(input_tensor)

License

Apache 2.0

Kunalsinghh/tms-marl-gnn-models

TMS2 - MARL Traffic Management Models

Multi-Agent Reinforcement Learning Models

Algorithms:

Features:

Model Description

Training Details

Usage

License

Kunalsinghh/
tms-marl-gnn-models