Downloads · 30 days
0
fpaissan/phinet_odl
phinet_odl is a machine learning model from fpaissan. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
Downloads · 30 days
0
Access
Public
Updated Apr 10, 2024
Repo size
5.1 MB
Likes
0
Public
Click a slice to open those files.
.tar5.1 MB · 100%
From the Hugging Face model README
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))