lagom icon indicating copy to clipboard operation
lagom copied to clipboard

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

Open zuoxingdong opened this issue 4 years ago • 0 comments

This will dramatically increase reusability and reduce boilerplate code.

e.g.

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

Networks:

  • Total parameters: trainable parameters, un-trainable parameters
    • Extract the number of such parameters
    • Extract a generator of such parameters
  • Gradient norm

Policy:

  • Entropy

zuoxingdong avatar Jun 14 '20 13:06 zuoxingdong