search

rmattson1008 / dqn_asteroid_probe

Probing various layers of DQN to assess "difficulty" of decisions.
MIT License
0 stars 0 forks source link

Keep track of training stats #3

Open rmattson1008 opened 1 year ago

rmattson1008 commented 1 year ago

Since this is a quick project, would be nice to skip a hefty tensorboard output. Realistically, I need the plots. However, I don't have to save anything except the final model (as full pth, or weights).

rmattson1008 commented 1 year ago

Just use the plot protocol in the dqn tutorial here, but save the plots rather than display.