search

Solacex / pytorch-implementation-of-DIRT-T

8 stars 2 forks source link

import pretrainedmodels #3

Closed PotatoThanh closed 3 years ago

PotatoThanh commented 3 years ago

Hi,

The code can't import pretrainedmodels module.

Thanks!

Solacex commented 3 years ago

Hello, the pretrainedmodels is the lib from https://github.com/Cadene/pretrained-models.pytorch, so I suggest you to install this package. Or replace it with other libs to import pretrained models, eg. torch.hub.

PotatoThanh commented 3 years ago

Thanks! btw I have one more question: How to create lmdb files? When I run python train.py it doesn't generate these files.

Solacex commented 3 years ago

Hello，as this is a very old code ... so I can remind of the detail of lmdb generation. I think you could try the to_lmdb.py or re-implement the dataloader modules.