Skip to content

Commit 4632c36

Browse files
SamuelGabrielfacebook-github-bot
authored andcommitted
Fix model paths in prior fitted networks (#2843)
Summary: Pull Request resolved: #2843 The model path for the main hebo model was misspecified in the prior fitted network code. I fixed that. Additionally, I removed the path that accepts a user prior as input, as that is not supported by the botorch code yet. Reviewed By: esantorella Differential Revision: D73131935 fbshipit-source-id: 3934106765a54886e6b3eb1deb719061a8d5166c
1 parent 78af542 commit 4632c36

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

botorch_community/models/utils/prior_fitted_network.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -28,16 +28,12 @@ class ModelPaths(Enum):
2828

2929
pfns4bo_hebo = (
3030
"https://github.com/automl/PFNs4BO/raw/refs/heads/main/pfns4bo"
31-
"/final_models/hebo_morebudget_9_unused_features_3_userpriorperdim2_8.pt.gz"
31+
"/final_models/model_hebo_morebudget_9_unused_features_3.pt.gz"
3232
)
3333
pfns4bo_bnn = (
3434
"https://github.com/automl/PFNs4BO/raw/refs/heads/main/pfns4bo"
3535
"/final_models/model_sampled_warp_simple_mlp_for_hpob_46.pt.gz"
3636
)
37-
pfns4bo_hebo_userprior = (
38-
"https://github.com/automl/PFNs4BO/raw/refs/heads/main/pfns4bo"
39-
"/final_models/hebo_morebudget_9_unused_features_3_userpriorperdim2_8.pt.gz"
40-
)
4137

4238

4339
def download_model(

0 commit comments

Comments
 (0)