Saving training metrics to a python variable

fastai
1

When you train a vision learner it prints train_loss/valid_loss etc. How do I get this into a python variable so that I can log this into wandb etc.

2

apparently there is an integration with wandb: fastai - Wandb.

Just need to figure out how to use it. Please delete this question if it is not particularly useful.

3

Hi! Did you actually manage to get the train_loss to a variable?

I found that loss_history = [float(x) for x in model.recorder.losses] gives me the valid_loss, but is there a way to get the train_loss?

Thanks!

4

Hey,
learn.recorder.values should provide that:
Screenshot from 2023-01-13 12-37-24

3 Likes
5

Haha damn, it was right there and I didn’t see it. Thanks a lot!

1 Like