semantic_memory_unit

http://www.dbpia.co.kr/journal/articleDetail?nodeId=NODE07613788

This is source code of the paper "Efficient external memory construction based on information gain" that experiments with House3D RoomNav task.

If you want to run this code, you have to install House3D Roomnav and you shold input your path to config.json

requirements

• python 3.6+
• pytorch 0.4.1

training

Before training, you have to configure gpu ids and number of cpu in Class Params in main.py

python main.py