Skip to content

text-classification issue  #76

@HARISHSENTHIL

Description

@HARISHSENTHIL

README.md checkpoint-1500 checkpoint-3000 checkpoint-4500 checkpoint-5500 checkpoint-7000 checkpoint-8500 checkpoint-9819 model.safetensors tokenizer_config.json training_args.bin
all_results.json checkpoint-2000 checkpoint-3500 checkpoint-500 checkpoint-6000 checkpoint-7500 checkpoint-9000 config.json special_tokens_map.json train_results.json vocab.txt
checkpoint-1000 checkpoint-2500 checkpoint-4000 checkpoint-5000 checkpoint-6500 checkpoint-8000 checkpoint-9500 eval_results.json tokenizer.json trainer_state.json

this is my check point file if try to access this. i am getting error { python3 qnli.py --ckpt /home/ollama/Apps/trak/qnli_model/model.safetensors --out /home/ollama/Apps/trak/trak_output }

Traceback (most recent call last):
File "/home/ollama/Apps/trak/trak/examples/qnli.py", line 160, in
model = init_model(args.ckpt, device)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/ollama/Apps/trak/trak/examples/qnli.py", line 135, in init_model
sd = ch.load(ckpt_path)
^^^^^^^^^^^^^^^^^^
File "/home/ollama/.pyenv/versions/3.11.0/lib/python3.11/site-packages/torch/serialization.py", line 1114, in load
return _legacy_load(
^^^^^^^^^^^^^
File "/home/ollama/.pyenv/versions/3.11.0/lib/python3.11/site-packages/torch/serialization.py", line 1338, in _legacy_load
magic_number = pickle_module.load(f, **pickle_load_args)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
_pickle.UnpicklingError: invalid load key, '\xe8'.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions