yrcong / STTran

Spatial-Temporal Transformer for Dynamic Scene Graph Generation, ICCV2021
MIT License
187 stars 34 forks source link

Trained model can't be unzipped #22

Closed yz09191 closed 2 years ago

yz09191 commented 2 years ago

I download your trained model, it is tar file and I have tried all unzip command in linux, but it can't be unzipped? I'm very grateful for your help!!

yrcong commented 2 years ago

I know it is somehow stupid. Actually .tar is already the snapshot. You can directly load it with torch.load('checkpoint.tar')