Skip to content

ma-shangao/rl_waypoint_mrta

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

rl_waypoint_mrta

Python

This project uses DRL to address waypoint following task allocation and planning for multi-robot systems.

1. Dependencies

This software relies on Python 3.9.

Please see the requirements.txt file for the details of dependencies. To install the dependencies, run the following command:

pip install -r requirements.txt

Otherwise, you can use the Dockerfile provided to build the environment. To build the docker image, run the following command:

docker build -t rlwaypointmrta:latest .

To run the docker image, run the following command:

docker run --rm -it  rlwaypointmrta:latest

The TSPLIB data files are from: http://elib.zib.de/pub/mp-testdata/tsp/tsplib/tsplib.html

The Pre-trained RL models for solving TSP problems are from: https://github.com/wouterkool/attention-learn-to-route

2. Usage

2.1. Training

python main.py --train

2.2. Evaluation

python main.py --eval --eval_dir trained_sessions/moe_mlp/rand_100-3/trained_model/batch31200.pt

2.3. Arguments

Arguments for training and evaluation that can be specified are defined in arg_parser.py. The docmentation of each argument is avaible by running the following command:

python main.py --help

2.4. Pre-trained models

Pre-trained models are available in the trained_sessions folder. The trained_model folder contains the trained model. Some relavent information about the training session is accessible using tesorboard. To run tensorboard, run the following command:

tensorboard --logdir=trained_sessions/$MODEL/$SESSION

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published