Skip to content

load-weights #41

Answered by Jmiceli17
vkaragounis asked this question in Q&A
Jun 3, 2024 · 3 comments · 1 reply
Discussion options

You must be logged in to vote

Hi!
I apologize, the load-weights parameter is currently unused. I have a ticket to remove deprecated config parameters but just haven't gotten to it yet.

After training, you should be seeing logs generated in a nn/<experiment_name> directory in the form of .pt files. These are the weights of your model. For example, if you're using ac-indepnedent-SUMO.py with parameter sharing, you will see a .pt file that was saved every number of max-cycles.

Those files can be loaded using the load_state_dict() function in torch. An example of this can be seen in sumo_analysis.py

Note though that it matters if you used parameter sharing or not during training. If you DID NOT, then the .pt files will ha…

Replies: 3 comments 1 reply

Comment options

You must be logged in to vote
0 replies
Answer selected by Jmiceli17
Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
1 reply
@Jmiceli17
Comment options

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants
Converted from issue

This discussion was converted from issue #36 on June 06, 2024 05:43.