fpaissan/phinet_odl · Hugging Face

from huggingface_hub import hf_hub_download
from micromind.networks import PhiNet

REPO_ID = "fpaissan/phinet_odl"
FILENAME = "phinet-odl.pth.tar"

path_state_dict = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)

model = PhiNet(
  (3, 128, 128),
  alpha=1.1,
  beta=0.75,
  t_zero=5,
  num_layers=8,
  h_swish=False,
  squeeze_excite=True,
  include_top=True,
  num_classes=1000,
  divisor=8,
  compatibility=False
)

model.load_state_dict(torch.load(path_state_dict))