ikuinen / CMIN_moment_retrieval

Cross-Modal Interaction Networks for Query-Based Moment Retrieval in Videos
86 stars 20 forks source link

Loading Pretrained Model #10

Closed AmeenAli closed 4 years ago

AmeenAli commented 4 years ago

Where can i Find in code the Pretrained Model loading? Because I can't see that this is implemented. Best

ikuinen commented 4 years ago

Since the training can be fast, you can easily achieve the results. You can load your trained model as follows:

self.model.load_state_dict(torch.load(model_path))

Where can i Find in code the Pretrained Model loading? Because I can't see that this is implemented. Best