GuanxingLu / ManiGaussian

[ECCV 2024] ManiGaussian: Dynamic Gaussian Splatting for Multi-task Robotic Manipulation
MIT License
152 stars 6 forks source link

Error on pickle.load() #17

Closed joonsu0109gh closed 1 month ago

joonsu0109gh commented 1 month ago

Is there any other option for the code below? It keeps raising an error.

ManiGaussian/third_party/YARR/yarr/runners/offline_train_runner.py line 391 d = pickle.load(f)

        # Here we fake a mini store (buffer)
        store = {store_element.name: {}
                 for store_element in self._storage_signature}
        if start_index % self._replay_capacity < end_index % self._replay_capacity:
            for i in range(start_index, end_index):
                    with open(join(self._save_dir, '%d.replay' % i), 'rb') as f:
                        d = pickle.load(f)  
                        # FIXME: _pickle.UnpicklingError: invalid load key, '\x00' # FIXME: replay file 9656
                        for k, v in d.items():
                            store[k][i] = v
joonsu0109gh commented 1 month ago

It is solved with several trials of the train code.

GuanxingLu commented 1 month ago

Thanks! Yes the data loading process implemented by YARR may be unstable, see #7 . Just try again can somehow solve the problem.