How to save the log which axis is each epoch not epoch's steps?

[hems111318136]

• I have marked all applicable categories:
  • exception-raising bug
  • RL algorithm bug
  • documentation request (i.e. "X is missing from the documentation.")
  • new feature request
  • design request (i.e. "X should be changed to Y.")
• I have visited the source website
• I have searched through the issue tracker for duplicates
• I have mentioned version numbers, operating system and environment, where applicable:

  import tianshou, gymnasium as gym, torch, numpy, sys
  print(tianshou.__version__, gym.__version__, torch.__version__, numpy.__version__, sys.version, sys.platform)

Hi,
I want to change the x-axis of the log during training to represent the epoch number instead of the step count of the epoch.
Is there already have the api to do so?

[MischaPanch]

Hi @hems111318136

Right now the logging is rather rigid and handled fully in the trainer.

We have been working on making the data collection and logging more flexible, so a series of pull requests will follow soon that will address this issue. For the moment, the only way to control logging for you would be to use a custom Trainer (inherit from BaseTrainer).

We expect to add these new features over the course of the next two months

[hems111318136]

Got it, thanks.