make project

DRLMAPF_A3C_RNN.py

@@ -11,8 +11,8 @@
 import GroupLock
 from ACNet import ACNet
 from Env_Builder import *
-from PRIMAL2Env import PRIMAL2Env
-from PRIMAL2Observer import PRIMAL2Observer
+from Primal2Env import PRIMAL2Env
+from Primal2Observer import PRIMAL2Observer
 from PRIMALObserver import PRIMALObserver
 dev_list = device_lib.list_local_devices()
 

README.md

@@ -1,15 +1,12 @@
 # PRIMAL_2: Pathfinding via Reinforcement and Imitation Multi_agent Learning - Lifelong
 
 ## Setting up Code
-- Install required dependancies from requirements.txt
-
+- `conda create --name primal python=3.7`
+- `conda activate primal`
+- `pip install -r requirements.txt`
 
 ## Running Code
-- Pick appropriate number of meta-agents and threads (`NUM_THREADS` = `Number of agents in each environment ` ) 
-- Name training run via `training_version` in `DRLMAPF_A3C_RNN.py`
-- Set desired RL-IL ratio and GIF frequencies 
-- Select appropriate Observer in `DRLMAPF_A3C_RNN.py` (PRIMAL2 by default) , but a new observer can also be constructed 
-- run DRLMAPF_A3C_RNN.py
+- `python pogema_wrapper.py`
 
 ## Key Files
 
@@ -21,19 +18,6 @@
 - `PRIMAL2Observer.py` - Defines the decentralized observation of each PRIMAL2 agent.
 - `PRIMALObserver.py` - Defines the decentralized observation of our previous work, PRIMAL. 
 - `Obsever_Builder.py` - The high level observation class
-
-
-## Other Links
-- fully trained PRIMAL2 model in one-shot environment -  https://www.dropbox.com/s/3nppkpy7psg0j5v/model_PRIMAL2_oneshot_3astarMaps.7z?dl=0
-- fully trained PRIMAL2 model in LMAPF environment - https://www.dropbox.com/s/6wjq2bje4mcjywj/model_PRIMAL2_continuous_3astarMaps.7z?dl=0
-
-
-## Authors
-
-[Mehul Damani](damanimehul24@gmail.com)
-
-[Zhiyao Luo](luozhiyao933@126.com)
-
-[Emerson Wenzel](emersonwenzel@gmail.com)
-
-[Guillaume Sartoretti](guillaume.sartoretti@gmail.com)
+- `pogema_wrapper.py` - Интегрирование PRIMAL2 в режиме LMAPF и среды POGEMA
+-  `midel_primal2_continuous` - Веса модели
+-  `render.svg` -  Сохранение результата работы программы

model_primal2_continuous/checkpoint

@@ -0,0 +1,2 @@
+model_checkpoint_path: "model-31300.cptk"
+all_model_checkpoint_paths: "model-31300.cptk"