{
  "type": "DagNetwork",
  "network.0": "{\"name\": \"network.0\", \"module\": {\"training\": false, \"_buffers\": {}, \"_backward_hooks\": {}, \"_forward_hooks\": {}, \"_forward_pre_hooks\": {}, \"_state_dict_hooks\": {}, \"_load_state_dict_pre_hooks\": {}, \"_modules\": {}, \"in_channels\": 3, \"out_channels\": 32, \"kernel_size\": [3, 3], \"stride\": [2, 2], \"padding\": [0, 0], \"dilation\": [1, 1], \"transposed\": false, \"output_padding\": [0, 0], \"groups\": 1, \"padding_mode\": \"zeros\", \"_padding_repeated_twice\": [0, 0, 0, 0], \"_parameters\": {\"weight\": {\"type\": \"Parameter\", \"shape\": [32, 3, 3, 3]}, \"bias\": {\"type\": \"Parameter\", \"shape\": [32]}}, \"obj_class\": \"Conv2d\", \"obj_module\": \"torch.nn.modules.conv\"}, \"module_type\": \"Conv2d\", \"parent_node_names\": [], \"child_node_names\": [\"network.1\"], \"output_to_child_idx\": {\"network.1\": [0]}, \"x_len\": 1, \"x_type\": 0, \"saved_args\": [], \"saved_kwargs\": {}, \"_arch_params_type\": \"DagNode\"}",
  "network.1": "{\"name\": \"network.1\", \"module\": {\"training\": false, \"_backward_hooks\": {}, \"_forward_hooks\": {}, \"_forward_pre_hooks\": {}, \"_state_dict_hooks\": {}, \"_load_state_dict_pre_hooks\": {}, \"_modules\": {}, \"num_features\": 32, \"eps\": 1e-05, \"momentum\": 0.1, \"affine\": true, \"track_running_stats\": true, \"_parameters\": {\"weight\": {\"type\": \"Parameter\", \"shape\": [32]}, \"bias\": {\"type\": \"Parameter\", \"shape\": [32]}}, \"obj_class\": \"BatchNorm2d\", \"obj_module\": \"torch.nn.modules.batchnorm\"}, \"module_type\": \"BatchNormalization\", \"parent_node_names\": [\"network.0\"], \"child_node_names\": [\"network.2\"], \"output_to_child_idx\": {\"network.2\": [0]}, \"x_len\": 1, \"x_type\": 0, \"saved_args\": [], \"saved_kwargs\": {}, \"_arch_params_type\": \"DagNode\"}",
  "network.2": "{\"name\": \"network.2\", \"module\": {\"training\": false, \"_parameters\": {}, \"_buffers\": {}, \"_backward_hooks\": {}, \"_forward_hooks\": {}, \"_forward_pre_hooks\": {}, \"_state_dict_hooks\": {}, \"_load_state_dict_pre_hooks\": {}, \"_modules\": {}, \"inplace\": false, \"obj_class\": \"ReLU\", \"obj_module\": \"torch.nn.modules.activation\"}, \"module_type\": \"ReLU\", \"parent_node_names\": [\"network.1\"], \"child_node_names\": [\"avg_pool\"], \"output_to_child_idx\": {\"avg_pool\": [0]}, \"x_len\": 1, \"x_type\": 0, \"saved_args\": [], \"saved_kwargs\": {}, \"_arch_params_type\": \"DagNode\"}",
  "avg_pool": "{\"name\": \"avg_pool\", \"module\": {\"training\": false, \"_parameters\": {}, \"_buffers\": {}, \"_backward_hooks\": {}, \"_forward_hooks\": {}, \"_forward_pre_hooks\": {}, \"_state_dict_hooks\": {}, \"_load_state_dict_pre_hooks\": {}, \"_modules\": {}, \"output_size\": [1, 1], \"obj_class\": \"AdaptiveAvgPool2d\", \"obj_module\": \"torch.nn.modules.pooling\"}, \"module_type\": \"AdaptiveAvgPool2d\", \"parent_node_names\": [\"network.2\"], \"child_node_names\": [\"f0.torch_func_flatten\"], \"output_to_child_idx\": {\"f0.torch_func_flatten\": [0]}, \"x_len\": 1, \"x_type\": 0, \"saved_args\": [], \"saved_kwargs\": {}, \"_arch_params_type\": \"DagNode\"}",
  "f0.torch_func_flatten": "{\"name\": \"f0.torch_func_flatten\", \"module\": {\"obj_class\": \"builtin_function_or_method\", \"obj_name\": \"flatten\"}, \"module_type\": \"torch_func_flatten\", \"parent_node_names\": [\"avg_pool\"], \"child_node_names\": [\"fc\"], \"output_to_child_idx\": {\"fc\": [0]}, \"x_len\": 1, \"x_type\": 0, \"saved_args\": [1], \"saved_kwargs\": {}, \"_arch_params_type\": \"DagNode\"}",
  "fc": "{\"name\": \"fc\", \"module\": {\"training\": false, \"_buffers\": {}, \"_backward_hooks\": {}, \"_forward_hooks\": {}, \"_forward_pre_hooks\": {}, \"_state_dict_hooks\": {}, \"_load_state_dict_pre_hooks\": {}, \"_modules\": {}, \"in_features\": 32, \"out_features\": 10, \"_parameters\": {\"weight\": {\"type\": \"Parameter\", \"shape\": [10, 32]}, \"bias\": {\"type\": \"Parameter\", \"shape\": [10]}}, \"obj_class\": \"Linear\", \"obj_module\": \"torch.nn.modules.linear\"}, \"module_type\": \"Linear\", \"parent_node_names\": [\"f0.torch_func_flatten\"], \"child_node_names\": [], \"output_to_child_idx\": {}, \"x_len\": 1, \"x_type\": 0, \"saved_args\": [], \"saved_kwargs\": {}, \"_arch_params_type\": \"DagNode\"}"
}