Unfortunately the docs page you've linked isn't as clear as it probably should be about differing behaviors/requirements/examples for different models. Still, from a quick glance at it & related source code, I think what you want to do should be possible in just a few lines of code.
Without seeing your code or the full error traceback, it's hard to guess what might be going wrong. (The error message excerpt you've shown implies some code may be trying a `logger` variable that's not defined.) Can you show your `EpochLogger` code & the full traceback you're getting?
- Gordon