thiagopbueno / model-aware-policy-optimization

MAPO: Model-Aware Policy Optimization algorithm
GNU General Public License v3.0
1 stars 0 forks source link

Feature/networks initialization #68

Closed 0xangelo closed 5 years ago

0xangelo commented 5 years ago

Add kernel initialization to fully connected layers.

Defaults taken from my PyTorch implementations, which in turn were based on OpenAI Baselines.

Closes #52

thiagopbueno commented 5 years ago

It's not clear to me that normc_initializer will be absolutely helpful... so it would be better if we could use it or not based on some config flag...

0xangelo commented 5 years ago

It's not clear to me that normc_initializer will be absolutely helpful... so it would be better if we could use it or not based on some config flag...

I'll add it to the model config.