zuoxingdong / lagom

lagom: A PyTorch infrastructure for rapid prototyping of reinforcement learning algorithms.
MIT License
373 stars 30 forks source link

Extract independent snapshot functions for metrics: network, RL, experiment, statistics etc. #209

Open zuoxingdong opened 4 years ago

zuoxingdong commented 4 years ago

This will dramatically increase reusability and reduce boilerplate code.

e.g.

snapshot_nn(net, ['total_parameters', 'trainable_parameters'])

Networks:

Policy: