/simple2transformer

A deep reinforcement learning method based on transformer for flexible job shop scheduling problems

Primary LanguagePython

simple2transformer

This is an implementation of the paper "A transformer-based deep reinforcement learning model for flexible job-shop scheduling problems with uncertainty"

Requirements

  • python >=3.6 (python=3.6 in our environment)
  • pytorch >= 1.10 (pytorch=1.10.2 in our environment)
  • the detailed requirements can be found in the requirements.txt file

Installation

pip install -r requirements.txt

Quick to start