Update hubconf.py
Browse files- hubconf.py +0 -3
hubconf.py
CHANGED
|
@@ -3,9 +3,6 @@ import torch
|
|
| 3 |
from pinn_electromagnetics.models import GeoNetHash, MaxwellPINN
|
| 4 |
|
| 5 |
def geonet(pretrained=False, map_location="cpu"):
|
| 6 |
-
"""
|
| 7 |
-
Loads the GeoNet model directly from a .pth file.
|
| 8 |
-
"""
|
| 9 |
model = GeoNetHash()
|
| 10 |
if pretrained:
|
| 11 |
file_path = hf_hub_download(
|
|
|
|
| 3 |
from pinn_electromagnetics.models import GeoNetHash, MaxwellPINN
|
| 4 |
|
| 5 |
def geonet(pretrained=False, map_location="cpu"):
|
|
|
|
|
|
|
|
|
|
| 6 |
model = GeoNetHash()
|
| 7 |
if pretrained:
|
| 8 |
file_path = hf_hub_download(
|