search

Vegetebird / StridedTransformer-Pose3D

[TMM 2022] Exploiting Temporal Contexts with Strided Transformer for 3D Human Pose Estimation
MIT License
335 stars 37 forks source link

PyTorch error #24

Closed Adithya2907 closed 1 year ago

Adithya2907 commented 1 year ago

I get the following error when I try to train the model using: python main.py --test --refine --reload --refine_reload --previous_dir checkpoint/pretrained

AttributeError: 'list' object has no attribute 'seek'. You can only torch.load from a file that is seekable. Please pre-load the data into a buffer like io.BytesIO and try to load from it instead.

I'm using Google Colab, with CUDA version of 11.2 and PyTorch version of 1.12, Python version 3.7

Vegetebird commented 1 year ago

Please check whether the pretrained model is in the "checkpoint/pretrained"