sqiud commited on
Commit
6524ea5
·
verified ·
1 Parent(s): 04b5fd3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +21 -1
README.md CHANGED
@@ -7,4 +7,24 @@ sdk: static
7
  pinned: false
8
  ---
9
 
10
- Edit this `README.md` markdown file to author your organization card.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
7
  pinned: false
8
  ---
9
 
10
+ Dataset stats: \
11
+ lat_mean = 39.951564548022596 \
12
+ lat_std = 0.0006361722351128644 \
13
+ lon_mean = -75.19150880602636 \
14
+ lon_std = 0.000611411894337979
15
+
16
+ The model can be loaded using:
17
+ ```
18
+ from huggingface_hub import hf_hub_download
19
+ import torch
20
+
21
+ # Specify the repository and the filename of the model you want to load
22
+ repo_id = "FinalProj5190/ImageToGPSproject-vit-base" # Replace with your repo name
23
+ filename = "resnet_gps_regressor_complete.pth"
24
+
25
+ model_path = hf_hub_download(repo_id=repo_id, filename=filename)
26
+
27
+ # Load the model using torch
28
+ model_test = torch.load(model_path)
29
+ model_test.eval() # Set the model to evaluation mode
30
+ ```