search

flowersteam / Grounding_LLMs_with_online_RL

We perform functional grounding of LLMs' knowledge in BabyAI-Text
MIT License
221 stars 24 forks source link

Fix/epoch logging #30

Closed HRezaei closed 3 months ago

HRezaei commented 3 months ago

Log values were initialized inside the epoch loop, thus only the last epoch was considered in the output. With this fix, log values of all epochs are collected, and then, their mean value is calculated and returned.

ClementRomac commented 3 months ago

Hey there, thanks for spotting this out!

Merging it right now :)

HRezaei commented 3 months ago

You're welcome! Thanks! :)