Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

get_net_param returns empty dictionary #49

Closed
fracapuano opened this issue Dec 17, 2022 · 1 comment
Closed

get_net_param returns empty dictionary #49

fracapuano opened this issue Dec 17, 2022 · 1 comment

Comments

@fracapuano
Copy link

Describe the bug
When retrieving network parameters with get_net_param I obtain an empty dict that I clearly cannot import as state_dict of a Pytorch model.

To Reproduce
Please provide a small script to reproduce the behavior:

from nats_bench import create
nats_api = create("NATS-tss-v1_0-3ffb9-simple", search_space="topology", fast_mode=True, verbose=False)
# sample random idx to simulate behavior
random_idx = nats_api.random()
# get corresponding architecture on CIFAR10
random_architecture = nats_api.get_net_config(index=random_idx, dataset="cifar10")
# retrieve network parameters according to main README.md
random_params = nats_api.get_net_param(index=random_idx, dataset="cifar10", seed=None)
print(random_params)
# prints {111: None}
print(next(iter(random_params.values())))
# prints None

OS: macOS Ventura 13.0.1
Python version: 3.10.8
Pytorch version: 1.13.0
Expected behavior
random_params should be a meaningful dictionary that I could use as params dict for a torch model

@D-X-Y
Copy link
Owner

D-X-Y commented Dec 20, 2022

The network parameters are contained in full archive files (please follow instructions at https://github.com/D-X-Y/NATS-Bench#preparation-and-download). Note that those files are pretty large.

@D-X-Y D-X-Y closed this as completed Jan 3, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants