runtime error

Exit code: 1. Reason: sed were not used when initializing BertModel: ['cls.predictions.bias', 'cls.predictions.transform.LayerNorm.bias', 'cls.seq_relationship.weight', 'cls.predictions.transform.dense.bias', 'cls.predictions.decoder.weight', 'cls.predictions.transform.dense.weight', 'cls.predictions.transform.LayerNorm.weight', 'cls.seq_relationship.bias'] - This IS expected if you are initializing BertModel from the checkpoint of a model trained on another task or with another architecture (e.g. initializing a BertForSequenceClassification model from a BertForPreTraining model). - This IS NOT expected if you are initializing BertModel from the checkpoint of a model that you expect to be exactly identical (initializing a BertForSequenceClassification model from a BertForSequenceClassification model). All the weights of BertModel were initialized from the model checkpoint at bert-base-uncased. If your task is similar to the task the model of the checkpoint was trained on, you can already use BertModel for predictions without further training. Traceback (most recent call last): File "/home/user/app/app.py", line 4, in <module> from load_model import entity_extractor File "/home/user/app/load_model.py", line 53, in <module> entity_extractor = BertEntityExtractor.load_trained_model() File "/home/user/app/EntityExtractor.py", line 211, in load_trained_model entity_extractor = BertEntityExtractor.load_saved(entity_extractor_path) File "/home/user/app/EntityExtractor.py", line 206, in load_saved extr.net.load_state_dict(torch.load('./TickerExtraction/entity_model2.pt',map_location=torch.device('cpu'))) #,map_location=torch.device('cpu') File "/usr/local/lib/python3.10/site-packages/torch/serialization.py", line 713, in load return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args) File "/usr/local/lib/python3.10/site-packages/torch/serialization.py", line 920, in _legacy_load magic_number = pickle_module.load(f, **pickle_load_args) _pickle.UnpicklingError: invalid load key, '<'.

Container logs: