Eclectic-Sheep / sheeprl

Distributed Reinforcement Learning accelerated by Lightning Fabric
https://eclecticsheep.ai
Apache License 2.0
322 stars 33 forks source link

Added toggle to train after game or after step in DreamerV3 #319

Closed LucaVendruscolo closed 1 month ago

LucaVendruscolo commented 2 months ago

Summary

Describe the purpose of the pull request, including:

added a config called "train_after_step" to sheeprl\sheeprl\configs\exp\dreamer_v3.yaml. It is set to true by default which doesn't affect the standard way the program runs but if it's set to false it will make the program wait until the end of the episode/game to train the algorithm with all the data gathered.

Type of Change

Please select the one relevant option below:

Checklist

Please confirm that the following tasks have been completed: