Eclectic-Sheep / sheeprl

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

Feature/dotdict #112

Closed belerico closed 9 months ago

belerico commented 9 months ago

Summary

This PR introduces the dotdict, a simple class that let the user access dict keys with the dot notation. This has been introduced to reduce the access time of the configs (omegaconf.DictConfig and similar) attributes. The main config, after reading it in the cli.py, will be transformed into a dotdict object . The class adds also the __setstate__ and __getstate__ so that the class is pickable (needed for torch.distributed, specially in the decoupled algorithms).

Type of Change

Please select the one relevant option below:

Checklist

Please confirm that the following tasks have been completed:

Thank you for your contribution! Once you have filled out this template, please ensure that you have assigned the appropriate reviewers and that all tests have passed.