Layout-Parser / layout-model-training

The scripts for training Detectron2-based Layout Models on popular layout analysis datasets
198 stars 54 forks source link

Config File Publaynet fastrcnn _pickle.UnpicklingError: invalid load key, '<'. error #19

Open Niteshiya opened 1 year ago

Niteshiya commented 1 year ago

Here is the log gived a pickle error . Also attached a error txt file for full output Traceback (most recent call last): File "/Users/niteshkumar/PycharmProjects/layout_parser/layout-model-training/tools/train_net.py", line 222, in <module> launch( File "/Users/niteshkumar/PycharmProjects/layout_parser/venv/lib/python3.9/site-packages/detectron2/engine/launch.py", line 84, in launch main_func(*args) File "/Users/niteshkumar/PycharmProjects/layout_parser/layout-model-training/tools/train_net.py", line 182, in main trainer.resume_or_load(resume=args.resume) File "/Users/niteshkumar/PycharmProjects/layout_parser/venv/lib/python3.9/site-packages/detectron2/engine/defaults.py", line 412, in resume_or_load self.checkpointer.resume_or_load(self.cfg.MODEL.WEIGHTS, resume=resume) File "/Users/niteshkumar/PycharmProjects/layout_parser/venv/lib/python3.9/site-packages/fvcore/common/checkpoint.py", line 227, in resume_or_load return self.load(path, checkpointables=[]) File "/Users/niteshkumar/PycharmProjects/layout_parser/venv/lib/python3.9/site-packages/detectron2/checkpoint/detection_checkpoint.py", line 62, in load ret = super().load(path, *args, **kwargs) File "/Users/niteshkumar/PycharmProjects/layout_parser/venv/lib/python3.9/site-packages/fvcore/common/checkpoint.py", line 155, in load checkpoint = self._load_file(path) File "/Users/niteshkumar/PycharmProjects/layout_parser/venv/lib/python3.9/site-packages/detectron2/checkpoint/detection_checkpoint.py", line 99, in _load_file loaded = self._torch_load(filename) File "/Users/niteshkumar/PycharmProjects/layout_parser/venv/lib/python3.9/site-packages/detectron2/checkpoint/detection_checkpoint.py", line 114, in _torch_load return super()._load_file(f) File "/Users/niteshkumar/PycharmProjects/layout_parser/venv/lib/python3.9/site-packages/fvcore/common/checkpoint.py", line 252, in _load_file return torch.load(f, map_location=torch.device("cpu")) File "/Users/niteshkumar/PycharmProjects/layout_parser/venv/lib/python3.9/site-packages/torch/serialization.py", line 815, in load return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args) File "/Users/niteshkumar/PycharmProjects/layout_parser/venv/lib/python3.9/site-packages/torch/serialization.py", line 1033, in _legacy_load magic_number = pickle_module.load(f, **pickle_load_args) _pickle.UnpicklingError: invalid load key, '<'.

error.txt