How to use the efficient pretrained model in a closed system with no internet access

[JunMa11]

Dear MONAI Team,

Thanks for the awesome work.

I want to use the pre-trained efficient model on a closed server without internet access.
I have manually downloaded and uploaded the pretrained models to the server.

My current code is

from monai.networks.nets import EfficientNetBN
model = EfficientNetBN("efficientnet-b0", pretrained=Train, spatial_dims=2, in_channels=3, num_classes=num_class).to(device)

outputs = model(inputs)

How should I modify the code to set the path to the pre-trained models and load them?

Kindest regards,
Jun

[Nic-Ma]

Hi @JunMa11 ,

Thanks for your feedback.
I feel this is an interesting topic, @yiheng-wang-nv could you please help think about it?
Maybe you can improve the pretrained logic your previously added to allow local path weights?

Thanks in advance.

[wyli]

the default dir is from torch.hub, you can put the downloaded .pth file in the default path https://github.com/pytorch/pytorch/blob/e1c449ff342ee00a4b25e1651bd08a41b7e9fda4/torch/hub.py#L568:

>>> import torch
>>> import os
>>> os.path.join(torch.hub.get_dir(), "checkpoints")

[Nic-Ma]

Good idea!

[JunMa11]

Thank you so much.