search

eleurent / rl-agents

Implementations of Reinforcement Learning and Planning algorithms
MIT License
553 stars 149 forks source link

How to change a default network structure? #81

Open lucalazzaroni opened 2 years ago

lucalazzaroni commented 2 years ago

Hi, I am currently working with this configuration, where you use different kinds of layers. How can I change the structure? Is it possible directly from your JSON file or should I put hands into the code? For example, I wonder if It would be possible to add more dense layers in output or additional attention layers. Thank you in advance for your response :)

KexianShen commented 2 years ago

take the output layers for example,

        "output_layer": {
            "type": "MultiLayerPerceptron",
            "layers": [64, 64, 64], // just add numbers here
            "reshape": false
        }
eleurent commented 2 years ago

Sorry the late reply, and thanks @KexianShen for jumping in :)

For more details, you can look at this function and subsequent classes to see how the config is parsed.