https://github.com/berenslab/retinal-rl/blob/89e0f6a3a7e4edf4910bc5186ba3b99107abbf0a/retinal_rl/analysis/plot.py#L322C18-L322C29 In this line probably the epoch number should be added to the filename
