search

yindaz / DeepCompletionRelease

Deep Depth Completion of a Single RGB-D Image
http://deepcompletion.cs.princeton.edu/
563 stars 141 forks source link

Convert pre-trained models to CPU before saving #36

Open danielcranston opened 4 years ago

danielcranston commented 4 years ago

I'm trying to load the pre-trained models (.t7 files) in PyTorch using https://github.com/clcarwin/convert_torch_to_pytorch. The error I run into points me to this issue in the same repo, suggesting that the issue is that the models were not converted to CPU before being saved.

In order to allow a broader community access to these models, I'd like to ask that you convert the models to CPU, re-save and provide these models as well.

My setup makes installing Lua and Torch quite cumbersome it seems, else I would have prepared these models and made a PR.

Thanks!

mac137 commented 4 years ago

Hi @danielcranston! Have you managed to convert the models to work on CPU? @yindaz would it be possible to provide the pre-trained models working on CPU?