While we’re at the topic of names: The params currently are just numbered. I must admit I’d think it’d be prettier if we used the state_dict names instead. What do you think?
Sure that sounds good. Since params are referenced by its numeric node id in Torch IR and we line-by-line translate, we still need association of numeric ID -> state_dict key name.
state_dict key name is available here as “full_attr” so you can use this name when creating Var.
full_attr_node_name
is the numeric ID corresponding to full_attr
. You need to add this mapping in outputs
variable.