aksub99 / MolDQN-pytorch

A PyTorch Implementation of "Optimization of Molecules via Deep Reinforcement Learning".
MIT License
75 stars 28 forks source link

Add support for bootstrap heads #4

Open aksub99 opened 4 years ago

aksub99 commented 4 years ago

The original paper https://arxiv.org/pdf/1810.08678.pdf experiments with bootstrap heads to improve the model's exploration tendency.