It bothered me that the two exisiting training scripts and methods were so alike, so I combined the 2 of them into 1, which can do both. I've added functionality for saving during benchmarks, which required some changes to Env.benchmark().
I've done my best on the docstrings, but please add comments here if something is unclear.
It bothered me that the two exisiting training scripts and methods were so alike, so I combined the 2 of them into 1, which can do both. I've added functionality for saving during benchmarks, which required some changes to
Env.benchmark()
.I've done my best on the docstrings, but please add comments here if something is unclear.