Fix RL Unplugged DM Lab 'Observation Pixels' issue

ManifoldRG / MultiNet

MIT License

10 stars 1 forks source link

Fix RL Unplugged DM Lab 'Observation Pixels' issue #18

Closed pranavguru closed 3 months ago

pranavguru commented 3 months ago

Decoding the DM Lab data which is stored in the TFRecord format results in the 'observation pixels' field not following the dimensions mentioned in the TFDS documentation - https://www.tensorflow.org/datasets/catalog/rlu_dmlab_explore_object_rewards_few

Fix this to make sure the DM Lab data is accurately loaded.

pranavguru commented 3 months ago

This issue is fixed. Use the tf.image_decode_jpeg module with the number of channels specified to correctly load TF Image fields in datasets stored in the TFRecord format.