Firstly, thank you for your amazing work!
I was trying to cite your work, Beyond Uniform Sampling: Offline Reinforcement Learning with Imbalanced Datasets as our baseline. We assume the DW, SoftDW and PowerDW+ https://github.com/Improbable-AI/dw-offline-rl/blob/d6c40291192a2d1ae368dc59b8a6afb9709b6131/CQL/JaxCQL/dataset_utils.py#L291 are the key branches. However, it seems the codes of load_sample_weights as well as the weights to be loaded are missing. Could you share the codes and weight? Thanks!
Best regards!
Firstly, thank you for your amazing work!
I was trying to cite your work, Beyond Uniform Sampling: Offline Reinforcement Learning with Imbalanced Datasets as our baseline. We assume the DW, SoftDW and PowerDW+ https://github.com/Improbable-AI/dw-offline-rl/blob/d6c40291192a2d1ae368dc59b8a6afb9709b6131/CQL/JaxCQL/dataset_utils.py#L291 are the key branches. However, it seems the codes of load_sample_weights as well as the weights to be loaded are missing. Could you share the codes and weight? Thanks!
Best regards!