# Attribute Error reported when loading training\_args.bin

**URL:** <https://discuss.huggingface.co/t/attribute-error-reported-when-loading-training-args-bin/17927>\
**Category:** 🤗Transformers\
**Created:** [May 13, 2022, 1:03pm UTC](https://discuss.huggingface.co/t/attribute-error-reported-when-loading-training-args-bin/17927 "2022-05-13T13:03:59Z")\
**Posts on this page:** 2\
**Page:** 1

<div class="post-metadata">

**Author:** ![charles0120](https://avatars.discourse-cdn.com/v4/letter/c/eb8c5e/32.png) [@charles0120](https://discuss.huggingface.co/u/charles0120)\
**Post date:** [May 13, 2022, 1:03pm UTC](https://discuss.huggingface.co/t/attribute-error-reported-when-loading-training-args-bin/17927/1 "2022-05-13T13:03:59Z")

</div>

I got attribute error when loading the trianing\_args.bin using:

```auto
training_args = torch.load("model_path/training_args.bin")

```

Error report:

```auto
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/cxsun/anaconda3/envs/SimCSE/lib/python3.8/site-packages/torch/serialization.py", line 607, in load
    return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
  File "/home/cxsun/anaconda3/envs/SimCSE/lib/python3.8/site-packages/torch/serialization.py", line 882, in _load
    result = unpickler.load()
  File "/home/cxsun/anaconda3/envs/SimCSE/lib/python3.8/site-packages/torch/serialization.py", line 875, in find_class
    return super().find_class(mod_name, name)
AttributeError: Can't get attribute 'OurTrainingArguments' on <module ' __main__' (built-in)>
>>> training_args = torch.load("result/my-sup-simcse-roberta-base/training_args.bin")Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/cxsun/anaconda3/envs/SimCSE/lib/python3.8/site-packages/torch/serialization.py", line 607, in load
    return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
  File "/home/cxsun/anaconda3/envs/SimCSE/lib/python3.8/site-packages/torch/serialization.py", line 882, in _load
    result = unpickler.load()
  File "/home/cxsun/anaconda3/envs/SimCSE/lib/python3.8/site-packages/torch/serialization.py", line 875, in find_class
    return super().find_class(mod_name, name)
AttributeError: Can't get attribute 'OurTrainingArguments' on <module ' __main__' (built-in)>
>>> torch.load("result/my-sup-simcse-roberta-base")
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/cxsun/anaconda3/envs/SimCSE/lib/python3.8/site-packages/torch/serialization.py", line 594, in load
    with _open_file_like(f, 'rb') as opened_file:
  File "/home/cxsun/anaconda3/envs/SimCSE/lib/python3.8/site-packages/torch/serialization.py", line 230, in _open_file_like
    return _open_file(name_or_buffer, mode)
  File "/home/cxsun/anaconda3/envs/SimCSE/lib/python3.8/site-packages/torch/serialization.py", line 211, in __init__
    super(_open_file, self). __init__ (open(name, mode))
IsADirectoryError: [Errno 21] Is a directory: 'result/my-sup-simcse-roberta-base'
>>> training_args = torch.load("result/my-sup-simcse-roberta-base/training_args.bin")
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/cxsun/anaconda3/envs/SimCSE/lib/python3.8/site-packages/torch/serialization.py", line 607, in load
    return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
  File "/home/cxsun/anaconda3/envs/SimCSE/lib/python3.8/site-packages/torch/serialization.py", line 882, in _load
    result = unpickler.load()
  File "/home/cxsun/anaconda3/envs/SimCSE/lib/python3.8/site-packages/torch/serialization.py", line 875, in find_class
    return super().find_class(mod_name, name)
AttributeError: Can't get attribute 'OurTrainingArguments' on <module ' __main__' (built-in)>

```

Anyone can help?

---

<div class="post-metadata">

**Author:** ![sgugger](https://sea2.discourse-cdn.com/hellohellohello/user_avatar/discuss.huggingface.co/sgugger/32/2291_2.png) [@sgugger](https://discuss.huggingface.co/u/sgugger)\
**Post date:** [May 13, 2022, 1:32pm UTC](https://discuss.huggingface.co/t/attribute-error-reported-when-loading-training-args-bin/17927/2 "2022-05-13T13:32:49Z")

</div>

It looks like you are using a subclass of `TrainingArguments`, and that this subclass is not defined in the script you’re trying to run.
