jmliu206 / LIC_TCM

MIT License
161 stars 24 forks source link

Inability to Extract Pretrained Model #8

Closed LoveDaisy closed 1 year ago

LoveDaisy commented 1 year ago

I recently downloaded the pretrained model provided in the Google Drive link included in the project's README. The downloaded file is a .tar file with a size of 920381364 bytes. Unfortunately, I am encountering an issue when attempting to extract the files using the tar command. The specific error message that appears is "This does not look like a tar archive", indicating a possible file corruption or damage.

To help in troubleshooting this issue, could you kindly provide a MD5 checksum for the original file? This would allow us to verify the integrity of the downloaded file and ascertain whether it is indeed damaged or not.

Additionally, it would be beneficial if an alternative download method could be provided, apart from Google Drive. This may help circumvent any potential issues with the current download process.

LoveDaisy commented 1 year ago

Hi everyone,

I just wanted to update you all regarding the file loading issue I was having with "mse_lambda_0.05.pth.tar". After discussing with the author via email, it turns out the solution was simpler than expected.

Instead of trying to extract the .tar file using a shell command, we can directly load the checkpoints using PyTorch, like so: torch.load("mse_lambda_0.05.pth.tar"). I've tested this and it works perfectly, with no error messages.

I wanted to share this solution here, in case anyone else runs into the same problem. A big thank you to the author for their prompt and helpful response.

I'll be closing this issue now, but if anyone has any further questions or related problems, feel free to comment or open a new issue.

Cheers,