search

numenta / nupic.embodied

GNU Affero General Public License v3.0
6 stars 6 forks source link

Small fixes on model loading #38

Closed vkakerbeck closed 3 years ago

vkakerbeck commented 3 years ago

I added the random ID to be specified in the load parameter so now it is not a boolean anymore but a string of the full run name. I also added the ID into the wandb name so it's easy to associate the wandb run with the trained model folder.

One thing we could think about is to add another flag to not save models (or not save when in debugging mode) so we don't create so many folders while debugging. Previously they were just overwritten but now with the random ID it quickly gets messy. What do you think?

I also added precomputed statistics files for the robot arm and atari breakout so it doesn't need to be calculated again.

lucasosouza commented 3 years ago

One thing we could think about is to add another flag to not save models (or not save when in debugging mode) so we don't create so many folders while debugging. Previously they were just overwritten but now with the random ID it quickly gets messy. What do you think?

Agree, that is probably a good idea. At least on debugging mode it shouldn't save, and an additional flag as well can be useful if you think it will help.