diff --git a/uncompiledmodels/driving_off_policy.onnx b/uncompiledmodels/driving_off_policy.onnx index 57cb0baf..334b3749 100644 Binary files a/uncompiledmodels/driving_off_policy.onnx and b/uncompiledmodels/driving_off_policy.onnx differ diff --git a/uncompiledmodels/driving_policy.onnx b/uncompiledmodels/driving_policy.onnx index 974a2e81..a0d75eb1 100644 Binary files a/uncompiledmodels/driving_policy.onnx and b/uncompiledmodels/driving_policy.onnx differ diff --git a/uncompiledmodels/driving_vision.onnx b/uncompiledmodels/driving_vision.onnx index 497a782c..8fa7e98c 100644 --- a/uncompiledmodels/driving_vision.onnx +++ b/uncompiledmodels/driving_vision.onnx @@ -1,16 +1,16 @@  -pytorch 2.10.0+cu130: - +pytorch 2.10.0+cu130: + img_to_copy node__to_copy"Cast* to JL namespace?: __main__.FlattenedVisionModel/_to_copy: aten._to_copy.defaultJ\ pkg.torch.onnx.class_hierarchy:['__main__.FlattenedVisionModel', 'aten._to_copy.default']J pkg.torch.onnx.fx_node%_to_copy : [num_users=1] = call_function[target=torch.ops.aten._to_copy.default](args = (%inputs_img,), kwargs = {dtype: torch.float16})J. -pkg.torch.onnx.name_scopes['', '_to_copy']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', '_to_copy']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - + big_img _to_copy_1node__to_copy_1"Cast* to @@ -18,10 +18,10 @@ _to_copy_1node__to_copy_1"Cast* namespaceA: __main__.FlattenedVisionModel/_to_copy_1: aten._to_copy.defaultJ\ pkg.torch.onnx.class_hierarchy:['__main__.FlattenedVisionModel', 'aten._to_copy.default']J pkg.torch.onnx.fx_node%_to_copy_1 : [num_users=1] = call_function[target=torch.ops.aten._to_copy.default](args = (%inputs_big_img,), kwargs = {dtype: torch.float16})J0 -pkg.torch.onnx.name_scopes['', '_to_copy_1']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', '_to_copy_1']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - + _to_copy _to_copy_1catnode_cat"Concat* @@ -29,8 +29,8 @@ _to_copy_1catnode_cat"Concat* namespace5: __main__.FlattenedVisionModel/cat: aten.cat.defaultJW pkg.torch.onnx.class_hierarchy5['__main__.FlattenedVisionModel', 'aten.cat.default']J pkg.torch.onnx.fx_nodez%cat : [num_users=1] = call_function[target=torch.ops.aten.cat.default](args = ([%_to_copy, %_to_copy_1], 1), kwargs = {})J) -pkg.torch.onnx.name_scopes ['', 'cat']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes ['', 'cat']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1)  cat @@ -39,9 +39,9 @@ _to_copy_1catnode_cat"Concat* pkg.torch.onnx.class_hierarchyZ['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'aten.sub.Tensor']J pkg.torch.onnx.fx_node%sub : [num_users=1] = call_function[target=torch.ops.aten.sub.Tensor](args = (%cat, %b_vision_model_vision__mean), kwargs = {})J@ pkg.torch.onnx.name_scopes"['', 'vision_model.vision', 'sub']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 184, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 184, in forward x = (x - self._mean) / self._std  sub @@ -50,11 +50,11 @@ _to_copy_1catnode_cat"Concat* pkg.torch.onnx.class_hierarchyZ['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'aten.div.Tensor']J pkg.torch.onnx.fx_node%div : [num_users=1] = call_function[target=torch.ops.aten.div.Tensor](args = (%sub, %b_vision_model_vision__std), kwargs = {})J@ pkg.torch.onnx.name_scopes"['', 'vision_model.vision', 'div']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 184, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 184, in forward x = (x - self._mean) / self._std - + div 2vision_model.vision._en.stem.0.reparam_conv.weight 0vision_model.vision._en.stem.0.reparam_conv.biasconv2d node_conv2d"Conv* @@ -66,39 +66,39 @@ _to_copy_1catnode_cat"Concat* namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stem: torch.nn.modules.container.Sequential/vision_model.vision._en.stem.0: timm.models.fastvit.MobileOneBlock/vision_model.vision._en.stem.0.reparam_conv: torch.nn.modules.conv.Conv2d/conv2d: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.MobileOneBlock', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%div, %p_vision_model_vision__en_stem_0_reparam_conv_weight, %p_vision_model_vision__en_stem_0_reparam_conv_bias, [2, 2], [1, 1]), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stem', 'vision_model.vision._en.stem.0', 'vision_model.vision._en.stem.0.reparam_conv', 'conv2d']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stem', 'vision_model.vision._en.stem.0', 'vision_model.vision._en.stem.0.reparam_conv', 'conv2d']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward return self.act(self.se(self.reparam_conv(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2dgelu node_gelu"Gelu* approximate"tanhJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stem: torch.nn.modules.container.Sequential/vision_model.vision._en.stem.0: timm.models.fastvit.MobileOneBlock/vision_model.vision._en.stem.0.act: timm.layers.activations.GELUTanh/gelu: aten.gelu.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.MobileOneBlock', 'timm.layers.activations.GELUTanh', 'aten.gelu.default']J pkg.torch.onnx.fx_nodez%gelu : [num_users=1] = call_function[target=torch.ops.aten.gelu.default](args = (%conv2d,), kwargs = {approximate: tanh})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stem', 'vision_model.vision._en.stem.0', 'vision_model.vision._en.stem.0.act', 'gelu']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stem', 'vision_model.vision._en.stem.0', 'vision_model.vision._en.stem.0.act', 'gelu']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward return self.act(self.se(self.reparam_conv(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') - + gelu 2vision_model.vision._en.stem.1.reparam_conv.weight 0vision_model.vision._en.stem.1.reparam_conv.biasconv2d_1 node_conv2d_1"Conv* @@ -110,39 +110,39 @@ _to_copy_1catnode_cat"Concat* namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stem: torch.nn.modules.container.Sequential/vision_model.vision._en.stem.1: timm.models.fastvit.MobileOneBlock/vision_model.vision._en.stem.1.reparam_conv: torch.nn.modules.conv.Conv2d/conv2d_1: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.MobileOneBlock', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_1 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%gelu, %p_vision_model_vision__en_stem_1_reparam_conv_weight, %p_vision_model_vision__en_stem_1_reparam_conv_bias, [2, 2], [1, 1], [1, 1], 64), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stem', 'vision_model.vision._en.stem.1', 'vision_model.vision._en.stem.1.reparam_conv', 'conv2d_1']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stem', 'vision_model.vision._en.stem.1', 'vision_model.vision._en.stem.1.reparam_conv', 'conv2d_1']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward return self.act(self.se(self.reparam_conv(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_1gelu_1 node_gelu_1"Gelu* approximate"tanhJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stem: torch.nn.modules.container.Sequential/vision_model.vision._en.stem.1: timm.models.fastvit.MobileOneBlock/vision_model.vision._en.stem.1.act: timm.layers.activations.GELUTanh/gelu_1: aten.gelu.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.MobileOneBlock', 'timm.layers.activations.GELUTanh', 'aten.gelu.default']J pkg.torch.onnx.fx_node~%gelu_1 : [num_users=1] = call_function[target=torch.ops.aten.gelu.default](args = (%conv2d_1,), kwargs = {approximate: tanh})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stem', 'vision_model.vision._en.stem.1', 'vision_model.vision._en.stem.1.act', 'gelu_1']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stem', 'vision_model.vision._en.stem.1', 'vision_model.vision._en.stem.1.act', 'gelu_1']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward return self.act(self.se(self.reparam_conv(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') - + gelu_1 2vision_model.vision._en.stem.2.reparam_conv.weight 0vision_model.vision._en.stem.2.reparam_conv.biasconv2d_2 node_conv2d_2"Conv* @@ -154,39 +154,39 @@ _to_copy_1catnode_cat"Concat* namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stem: torch.nn.modules.container.Sequential/vision_model.vision._en.stem.2: timm.models.fastvit.MobileOneBlock/vision_model.vision._en.stem.2.reparam_conv: torch.nn.modules.conv.Conv2d/conv2d_2: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.MobileOneBlock', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_2 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%gelu_1, %p_vision_model_vision__en_stem_2_reparam_conv_weight, %p_vision_model_vision__en_stem_2_reparam_conv_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stem', 'vision_model.vision._en.stem.2', 'vision_model.vision._en.stem.2.reparam_conv', 'conv2d_2']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stem', 'vision_model.vision._en.stem.2', 'vision_model.vision._en.stem.2.reparam_conv', 'conv2d_2']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward return self.act(self.se(self.reparam_conv(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_2gelu_2 node_gelu_2"Gelu* approximate"tanhJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stem: torch.nn.modules.container.Sequential/vision_model.vision._en.stem.2: timm.models.fastvit.MobileOneBlock/vision_model.vision._en.stem.2.act: timm.layers.activations.GELUTanh/gelu_2: aten.gelu.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.MobileOneBlock', 'timm.layers.activations.GELUTanh', 'aten.gelu.default']J pkg.torch.onnx.fx_node~%gelu_2 : [num_users=1] = call_function[target=torch.ops.aten.gelu.default](args = (%conv2d_2,), kwargs = {approximate: tanh})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stem', 'vision_model.vision._en.stem.2', 'vision_model.vision._en.stem.2.act', 'gelu_2']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stem', 'vision_model.vision._en.stem.2', 'vision_model.vision._en.stem.2.act', 'gelu_2']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward return self.act(self.se(self.reparam_conv(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') - + gelu_2 Ivision_model.vision._en.stages.0.blocks.0.token_mixer.reparam_conv.weight Gvision_model.vision._en.stages.0.blocks.0.token_mixer.reparam_conv.biasconv2d_3 node_conv2d_3"Conv* @@ -198,24 +198,24 @@ Gvision_model.vision._en.stages.0.blocks.0.token_mixer.reparam_conv.biasconv2d namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.0: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.0.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.0.blocks.0: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.0.blocks.0.token_mixer: timm.models.fastvit.RepMixer/vision_model.vision._en.stages.0.blocks.0.token_mixer.reparam_conv: torch.nn.modules.conv.Conv2d/conv2d_3: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.RepMixer', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_3 : [num_users=2] = call_function[target=torch.ops.aten.conv2d.default](args = (%gelu_2, %p_vision_model_vision__en_stages_0_blocks_0_token_mixer_reparam_conv_weight, %p_vision_model_vision__en_stages_0_blocks_0_token_mixer_reparam_conv_bias, [1, 1], [1, 1], [1, 1], 64), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.0', 'vision_model.vision._en.stages.0.blocks.0.token_mixer', 'vision_model.vision._en.stages.0.blocks.0.token_mixer.reparam_conv', 'conv2d_3']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.0', 'vision_model.vision._en.stages.0.blocks.0.token_mixer', 'vision_model.vision._en.stages.0.blocks.0.token_mixer.reparam_conv', 'conv2d_3']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1001, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1001, in forward x = self.token_mixer(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward x = self.reparam_conv(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_3 >vision_model.vision._en.stages.0.blocks.0.mlp.conv.conv.weight Cvision_model.vision._en.stages.0.blocks.0.mlp.conv.conv.weight_biasgetitem node_Conv_182"Conv* @@ -228,26 +228,26 @@ Cvision_model.vision._en.stages.0.blocks.0.mlp.conv.conv.weight_biasgetitem n !pkg.onnxscript.rewriter.rule_name1FuseBatchNormIntoConv, RemoveOptionalBiasFromConvJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'timm.layers.conv_bn_act.ConvNormAct', 'timm.layers.norm_act.BatchNormAct2d', 'aten._native_batch_norm_legit_no_training.default']J pkg.torch.onnx.fx_node%_native_batch_norm_legit_no_training : [num_users=1] = call_function[target=torch.ops.aten._native_batch_norm_legit_no_training.default](args = (%conv2d_4, %p_vision_model_vision__en_stages_0_blocks_0_mlp_conv_bn_weight, %p_vision_model_vision__en_stages_0_blocks_0_mlp_conv_bn_bias, %b_vision_model_vision__en_stages_0_blocks_0_mlp_conv_bn_running_mean, %b_vision_model_vision__en_stages_0_blocks_0_mlp_conv_bn_running_var, 0.1, 1e-05), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.0', 'vision_model.vision._en.stages.0.blocks.0.mlp', 'vision_model.vision._en.stages.0.blocks.0.mlp.conv', 'vision_model.vision._en.stages.0.blocks.0.mlp.conv.bn', '_native_batch_norm_legit_no_training']J -pkg.torch.onnx.stack_trace File "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.0', 'vision_model.vision._en.stages.0.blocks.0.mlp', 'vision_model.vision._en.stages.0.blocks.0.mlp.conv', 'vision_model.vision._en.stages.0.blocks.0.mlp.conv.bn', '_native_batch_norm_legit_no_training']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 814, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 814, in forward x = self.conv(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/conv_bn_act.py", line 94, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/conv_bn_act.py", line 94, in forward x = self.bn(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward x = F.batch_norm( - + getitem 8vision_model.vision._en.stages.0.blocks.0.mlp.fc1.weight 6vision_model.vision._en.stages.0.blocks.0.mlp.fc1.biasconv2d_5 node_conv2d_5"Conv* @@ -259,47 +259,47 @@ Cvision_model.vision._en.stages.0.blocks.0.mlp.conv.conv.weight_biasgetitem n namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.0: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.0.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.0.blocks.0: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.0.blocks.0.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.0.blocks.0.mlp.fc1: torch.nn.modules.conv.Conv2d/conv2d_5: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_5 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%getitem, %p_vision_model_vision__en_stages_0_blocks_0_mlp_fc1_weight, %p_vision_model_vision__en_stages_0_blocks_0_mlp_fc1_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.0', 'vision_model.vision._en.stages.0.blocks.0.mlp', 'vision_model.vision._en.stages.0.blocks.0.mlp.fc1', 'conv2d_5']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.0', 'vision_model.vision._en.stages.0.blocks.0.mlp', 'vision_model.vision._en.stages.0.blocks.0.mlp.fc1', 'conv2d_5']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward x = self.fc1(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_5gelu_3 node_gelu_3"Gelu* approximate"tanhJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.0: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.0.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.0.blocks.0: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.0.blocks.0.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.0.blocks.0.mlp.act: timm.layers.activations.GELUTanh/gelu_3: aten.gelu.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'timm.layers.activations.GELUTanh', 'aten.gelu.default']J pkg.torch.onnx.fx_node~%gelu_3 : [num_users=1] = call_function[target=torch.ops.aten.gelu.default](args = (%conv2d_5,), kwargs = {approximate: tanh})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.0', 'vision_model.vision._en.stages.0.blocks.0.mlp', 'vision_model.vision._en.stages.0.blocks.0.mlp.act', 'gelu_3']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.0', 'vision_model.vision._en.stages.0.blocks.0.mlp', 'vision_model.vision._en.stages.0.blocks.0.mlp.act', 'gelu_3']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, in forward x = self.act(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') - + gelu_3 8vision_model.vision._en.stages.0.blocks.0.mlp.fc2.weight 6vision_model.vision._en.stages.0.blocks.0.mlp.fc2.biasconv2d_6 node_conv2d_6"Conv* @@ -311,64 +311,64 @@ Cvision_model.vision._en.stages.0.blocks.0.mlp.conv.conv.weight_biasgetitem n namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.0: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.0.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.0.blocks.0: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.0.blocks.0.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.0.blocks.0.mlp.fc2: torch.nn.modules.conv.Conv2d/conv2d_6: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_6 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%clone, %p_vision_model_vision__en_stages_0_blocks_0_mlp_fc2_weight, %p_vision_model_vision__en_stages_0_blocks_0_mlp_fc2_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.0', 'vision_model.vision._en.stages.0.blocks.0.mlp', 'vision_model.vision._en.stages.0.blocks.0.mlp.fc2', 'conv2d_6']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.0', 'vision_model.vision._en.stages.0.blocks.0.mlp', 'vision_model.vision._en.stages.0.blocks.0.mlp.fc2', 'conv2d_6']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward x = self.fc2(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_6 ;vision_model.vision._en.stages.0.blocks.0.layer_scale.gammamulnode_mul"MulJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.0: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.0.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.0.blocks.0: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.0.blocks.0.layer_scale: timm.models.fastvit.LayerScale2d/mul: aten.mul.TensorJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.LayerScale2d', 'aten.mul.Tensor']J pkg.torch.onnx.fx_node%mul : [num_users=1] = call_function[target=torch.ops.aten.mul.Tensor](args = (%clone_1, %p_vision_model_vision__en_stages_0_blocks_0_layer_scale_gamma), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.0', 'vision_model.vision._en.stages.0.blocks.0.layer_scale', 'mul']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.0', 'vision_model.vision._en.stages.0.blocks.0.layer_scale', 'mul']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 646, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 646, in forward return x.mul_(self.gamma) if self.inplace else x * self.gamma - + conv2d_3 muladdnode_add"AddJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.0: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.0.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.0.blocks.0: timm.models.fastvit.RepMixerBlock/add: aten.add.TensorJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'aten.add.Tensor']J pkg.torch.onnx.fx_nodem%add : [num_users=1] = call_function[target=torch.ops.aten.add.Tensor](args = (%conv2d_3, %mul), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.0', 'add']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.0', 'add']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - + add Ivision_model.vision._en.stages.0.blocks.1.token_mixer.reparam_conv.weight Gvision_model.vision._en.stages.0.blocks.1.token_mixer.reparam_conv.biasconv2d_7 node_conv2d_7"Conv* @@ -380,24 +380,24 @@ Gvision_model.vision._en.stages.0.blocks.1.token_mixer.reparam_conv.biasconv2d namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.0: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.0.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.0.blocks.1: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.0.blocks.1.token_mixer: timm.models.fastvit.RepMixer/vision_model.vision._en.stages.0.blocks.1.token_mixer.reparam_conv: torch.nn.modules.conv.Conv2d/conv2d_7: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.RepMixer', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_7 : [num_users=2] = call_function[target=torch.ops.aten.conv2d.default](args = (%add, %p_vision_model_vision__en_stages_0_blocks_1_token_mixer_reparam_conv_weight, %p_vision_model_vision__en_stages_0_blocks_1_token_mixer_reparam_conv_bias, [1, 1], [1, 1], [1, 1], 64), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.1', 'vision_model.vision._en.stages.0.blocks.1.token_mixer', 'vision_model.vision._en.stages.0.blocks.1.token_mixer.reparam_conv', 'conv2d_7']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.1', 'vision_model.vision._en.stages.0.blocks.1.token_mixer', 'vision_model.vision._en.stages.0.blocks.1.token_mixer.reparam_conv', 'conv2d_7']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1001, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1001, in forward x = self.token_mixer(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward x = self.reparam_conv(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_7 >vision_model.vision._en.stages.0.blocks.1.mlp.conv.conv.weight Cvision_model.vision._en.stages.0.blocks.1.mlp.conv.conv.weight_bias getitem_3 node_Conv_184"Conv* @@ -410,26 +410,26 @@ Cvision_model.vision._en.stages.0.blocks.1.mlp.conv.conv.weight_bias getitem_3 !pkg.onnxscript.rewriter.rule_name1FuseBatchNormIntoConv, RemoveOptionalBiasFromConvJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'timm.layers.conv_bn_act.ConvNormAct', 'timm.layers.norm_act.BatchNormAct2d', 'aten._native_batch_norm_legit_no_training.default']J pkg.torch.onnx.fx_node%_native_batch_norm_legit_no_training_1 : [num_users=1] = call_function[target=torch.ops.aten._native_batch_norm_legit_no_training.default](args = (%conv2d_8, %p_vision_model_vision__en_stages_0_blocks_1_mlp_conv_bn_weight, %p_vision_model_vision__en_stages_0_blocks_1_mlp_conv_bn_bias, %b_vision_model_vision__en_stages_0_blocks_1_mlp_conv_bn_running_mean, %b_vision_model_vision__en_stages_0_blocks_1_mlp_conv_bn_running_var, 0.1, 1e-05), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.1', 'vision_model.vision._en.stages.0.blocks.1.mlp', 'vision_model.vision._en.stages.0.blocks.1.mlp.conv', 'vision_model.vision._en.stages.0.blocks.1.mlp.conv.bn', '_native_batch_norm_legit_no_training_1']J -pkg.torch.onnx.stack_trace File "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.1', 'vision_model.vision._en.stages.0.blocks.1.mlp', 'vision_model.vision._en.stages.0.blocks.1.mlp.conv', 'vision_model.vision._en.stages.0.blocks.1.mlp.conv.bn', '_native_batch_norm_legit_no_training_1']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 814, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 814, in forward x = self.conv(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/conv_bn_act.py", line 94, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/conv_bn_act.py", line 94, in forward x = self.bn(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward x = F.batch_norm( - + getitem_3 8vision_model.vision._en.stages.0.blocks.1.mlp.fc1.weight 6vision_model.vision._en.stages.0.blocks.1.mlp.fc1.biasconv2d_9 node_conv2d_9"Conv* @@ -441,47 +441,47 @@ Cvision_model.vision._en.stages.0.blocks.1.mlp.conv.conv.weight_bias getitem_3 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.0: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.0.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.0.blocks.1: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.0.blocks.1.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.0.blocks.1.mlp.fc1: torch.nn.modules.conv.Conv2d/conv2d_9: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_9 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%getitem_3, %p_vision_model_vision__en_stages_0_blocks_1_mlp_fc1_weight, %p_vision_model_vision__en_stages_0_blocks_1_mlp_fc1_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.1', 'vision_model.vision._en.stages.0.blocks.1.mlp', 'vision_model.vision._en.stages.0.blocks.1.mlp.fc1', 'conv2d_9']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.1', 'vision_model.vision._en.stages.0.blocks.1.mlp', 'vision_model.vision._en.stages.0.blocks.1.mlp.fc1', 'conv2d_9']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward x = self.fc1(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_9gelu_4 node_gelu_4"Gelu* approximate"tanhJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.0: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.0.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.0.blocks.1: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.0.blocks.1.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.0.blocks.1.mlp.act: timm.layers.activations.GELUTanh/gelu_4: aten.gelu.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'timm.layers.activations.GELUTanh', 'aten.gelu.default']J pkg.torch.onnx.fx_node~%gelu_4 : [num_users=1] = call_function[target=torch.ops.aten.gelu.default](args = (%conv2d_9,), kwargs = {approximate: tanh})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.1', 'vision_model.vision._en.stages.0.blocks.1.mlp', 'vision_model.vision._en.stages.0.blocks.1.mlp.act', 'gelu_4']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.1', 'vision_model.vision._en.stages.0.blocks.1.mlp', 'vision_model.vision._en.stages.0.blocks.1.mlp.act', 'gelu_4']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, in forward x = self.act(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') - + gelu_4 8vision_model.vision._en.stages.0.blocks.1.mlp.fc2.weight 6vision_model.vision._en.stages.0.blocks.1.mlp.fc2.bias conv2d_10node_conv2d_10"Conv* @@ -493,66 +493,66 @@ Cvision_model.vision._en.stages.0.blocks.1.mlp.conv.conv.weight_bias getitem_3 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.0: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.0.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.0.blocks.1: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.0.blocks.1.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.0.blocks.1.mlp.fc2: torch.nn.modules.conv.Conv2d/conv2d_10: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_10 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%clone_2, %p_vision_model_vision__en_stages_0_blocks_1_mlp_fc2_weight, %p_vision_model_vision__en_stages_0_blocks_1_mlp_fc2_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.1', 'vision_model.vision._en.stages.0.blocks.1.mlp', 'vision_model.vision._en.stages.0.blocks.1.mlp.fc2', 'conv2d_10']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.1', 'vision_model.vision._en.stages.0.blocks.1.mlp', 'vision_model.vision._en.stages.0.blocks.1.mlp.fc2', 'conv2d_10']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward x = self.fc2(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_10 ;vision_model.vision._en.stages.0.blocks.1.layer_scale.gammamul_1 node_mul_1"MulJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.0: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.0.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.0.blocks.1: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.0.blocks.1.layer_scale: timm.models.fastvit.LayerScale2d/mul_1: aten.mul.TensorJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.LayerScale2d', 'aten.mul.Tensor']J pkg.torch.onnx.fx_node%mul_1 : [num_users=1] = call_function[target=torch.ops.aten.mul.Tensor](args = (%clone_3, %p_vision_model_vision__en_stages_0_blocks_1_layer_scale_gamma), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.1', 'vision_model.vision._en.stages.0.blocks.1.layer_scale', 'mul_1']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.1', 'vision_model.vision._en.stages.0.blocks.1.layer_scale', 'mul_1']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 646, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 646, in forward return x.mul_(self.gamma) if self.inplace else x * self.gamma - + conv2d_7 mul_1add_1 node_add_1"AddJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.0: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.0.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.0.blocks.1: timm.models.fastvit.RepMixerBlock/add_1: aten.add.TensorJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'aten.add.Tensor']J pkg.torch.onnx.fx_nodeq%add_1 : [num_users=1] = call_function[target=torch.ops.aten.add.Tensor](args = (%conv2d_7, %mul_1), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.1', 'add_1']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.0', 'vision_model.vision._en.stages.0.blocks', 'vision_model.vision._en.stages.0.blocks.1', 'add_1']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - + add_1 Fvision_model.vision._en.stages.1.downsample.proj.0.reparam_conv.weight Dvision_model.vision._en.stages.1.downsample.proj.0.reparam_conv.bias conv2d_11node_conv2d_11"Conv* @@ -564,24 +564,24 @@ Dvision_model.vision._en.stages.1.downsample.proj.0.reparam_conv.bias conv2d_11 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.1: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.1.downsample: timm.models.fastvit.PatchEmbed/vision_model.vision._en.stages.1.downsample.proj: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.1.downsample.proj.0: timm.models.fastvit.ReparamLargeKernelConv/vision_model.vision._en.stages.1.downsample.proj.0.reparam_conv: torch.nn.modules.conv.Conv2d/conv2d_11: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'timm.models.fastvit.PatchEmbed', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.ReparamLargeKernelConv', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_11 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%add_1, %p_vision_model_vision__en_stages_1_downsample_proj_0_reparam_conv_weight, %p_vision_model_vision__en_stages_1_downsample_proj_0_reparam_conv_bias, [2, 2], [3, 3], [1, 1], 64), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.downsample', 'vision_model.vision._en.stages.1.downsample.proj', 'vision_model.vision._en.stages.1.downsample.proj.0', 'vision_model.vision._en.stages.1.downsample.proj.0.reparam_conv', 'conv2d_11']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.downsample', 'vision_model.vision._en.stages.1.downsample.proj', 'vision_model.vision._en.stages.1.downsample.proj.0', 'vision_model.vision._en.stages.1.downsample.proj.0.reparam_conv', 'conv2d_11']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1162, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1162, in forward x = self.downsample(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 628, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 628, in forward x = self.proj(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 376, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 376, in forward out = self.reparam_conv(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_11 Fvision_model.vision._en.stages.1.downsample.proj.1.reparam_conv.weight Dvision_model.vision._en.stages.1.downsample.proj.1.reparam_conv.bias conv2d_12node_conv2d_12"Conv* @@ -593,47 +593,47 @@ Dvision_model.vision._en.stages.1.downsample.proj.1.reparam_conv.bias conv2d_12 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.1: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.1.downsample: timm.models.fastvit.PatchEmbed/vision_model.vision._en.stages.1.downsample.proj: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.1.downsample.proj.1: timm.models.fastvit.MobileOneBlock/vision_model.vision._en.stages.1.downsample.proj.1.reparam_conv: torch.nn.modules.conv.Conv2d/conv2d_12: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'timm.models.fastvit.PatchEmbed', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.MobileOneBlock', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_12 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%conv2d_11, %p_vision_model_vision__en_stages_1_downsample_proj_1_reparam_conv_weight, %p_vision_model_vision__en_stages_1_downsample_proj_1_reparam_conv_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.downsample', 'vision_model.vision._en.stages.1.downsample.proj', 'vision_model.vision._en.stages.1.downsample.proj.1', 'vision_model.vision._en.stages.1.downsample.proj.1.reparam_conv', 'conv2d_12']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.downsample', 'vision_model.vision._en.stages.1.downsample.proj', 'vision_model.vision._en.stages.1.downsample.proj.1', 'vision_model.vision._en.stages.1.downsample.proj.1.reparam_conv', 'conv2d_12']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1162, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1162, in forward x = self.downsample(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 628, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 628, in forward x = self.proj(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward return self.act(self.se(self.reparam_conv(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_12gelu_5 node_gelu_5"Gelu* approximate"tanhJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.1: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.1.downsample: timm.models.fastvit.PatchEmbed/vision_model.vision._en.stages.1.downsample.proj: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.1.downsample.proj.1: timm.models.fastvit.MobileOneBlock/vision_model.vision._en.stages.1.downsample.proj.1.act: timm.layers.activations.GELUTanh/gelu_5: aten.gelu.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'timm.models.fastvit.PatchEmbed', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.MobileOneBlock', 'timm.layers.activations.GELUTanh', 'aten.gelu.default']J pkg.torch.onnx.fx_node%gelu_5 : [num_users=1] = call_function[target=torch.ops.aten.gelu.default](args = (%conv2d_12,), kwargs = {approximate: tanh})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.downsample', 'vision_model.vision._en.stages.1.downsample.proj', 'vision_model.vision._en.stages.1.downsample.proj.1', 'vision_model.vision._en.stages.1.downsample.proj.1.act', 'gelu_5']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.downsample', 'vision_model.vision._en.stages.1.downsample.proj', 'vision_model.vision._en.stages.1.downsample.proj.1', 'vision_model.vision._en.stages.1.downsample.proj.1.act', 'gelu_5']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1162, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1162, in forward x = self.downsample(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 628, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 628, in forward x = self.proj(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward return self.act(self.se(self.reparam_conv(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') - + gelu_5 Ivision_model.vision._en.stages.1.blocks.0.token_mixer.reparam_conv.weight Gvision_model.vision._en.stages.1.blocks.0.token_mixer.reparam_conv.bias conv2d_13node_conv2d_13"Conv* @@ -645,24 +645,24 @@ Gvision_model.vision._en.stages.1.blocks.0.token_mixer.reparam_conv.bias conv2d namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.1: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.1.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.1.blocks.0: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.1.blocks.0.token_mixer: timm.models.fastvit.RepMixer/vision_model.vision._en.stages.1.blocks.0.token_mixer.reparam_conv: torch.nn.modules.conv.Conv2d/conv2d_13: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.RepMixer', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_13 : [num_users=2] = call_function[target=torch.ops.aten.conv2d.default](args = (%gelu_5, %p_vision_model_vision__en_stages_1_blocks_0_token_mixer_reparam_conv_weight, %p_vision_model_vision__en_stages_1_blocks_0_token_mixer_reparam_conv_bias, [1, 1], [1, 1], [1, 1], 128), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.0', 'vision_model.vision._en.stages.1.blocks.0.token_mixer', 'vision_model.vision._en.stages.1.blocks.0.token_mixer.reparam_conv', 'conv2d_13']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.0', 'vision_model.vision._en.stages.1.blocks.0.token_mixer', 'vision_model.vision._en.stages.1.blocks.0.token_mixer.reparam_conv', 'conv2d_13']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1001, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1001, in forward x = self.token_mixer(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward x = self.reparam_conv(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_13 >vision_model.vision._en.stages.1.blocks.0.mlp.conv.conv.weight Cvision_model.vision._en.stages.1.blocks.0.mlp.conv.conv.weight_bias getitem_6 node_Conv_186"Conv* @@ -675,26 +675,26 @@ Cvision_model.vision._en.stages.1.blocks.0.mlp.conv.conv.weight_bias getitem_6 !pkg.onnxscript.rewriter.rule_name1FuseBatchNormIntoConv, RemoveOptionalBiasFromConvJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'timm.layers.conv_bn_act.ConvNormAct', 'timm.layers.norm_act.BatchNormAct2d', 'aten._native_batch_norm_legit_no_training.default']J pkg.torch.onnx.fx_node%_native_batch_norm_legit_no_training_2 : [num_users=1] = call_function[target=torch.ops.aten._native_batch_norm_legit_no_training.default](args = (%conv2d_14, %p_vision_model_vision__en_stages_1_blocks_0_mlp_conv_bn_weight, %p_vision_model_vision__en_stages_1_blocks_0_mlp_conv_bn_bias, %b_vision_model_vision__en_stages_1_blocks_0_mlp_conv_bn_running_mean, %b_vision_model_vision__en_stages_1_blocks_0_mlp_conv_bn_running_var, 0.1, 1e-05), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.0', 'vision_model.vision._en.stages.1.blocks.0.mlp', 'vision_model.vision._en.stages.1.blocks.0.mlp.conv', 'vision_model.vision._en.stages.1.blocks.0.mlp.conv.bn', '_native_batch_norm_legit_no_training_2']J -pkg.torch.onnx.stack_trace File "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.0', 'vision_model.vision._en.stages.1.blocks.0.mlp', 'vision_model.vision._en.stages.1.blocks.0.mlp.conv', 'vision_model.vision._en.stages.1.blocks.0.mlp.conv.bn', '_native_batch_norm_legit_no_training_2']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 814, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 814, in forward x = self.conv(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/conv_bn_act.py", line 94, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/conv_bn_act.py", line 94, in forward x = self.bn(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward x = F.batch_norm( - + getitem_6 8vision_model.vision._en.stages.1.blocks.0.mlp.fc1.weight 6vision_model.vision._en.stages.1.blocks.0.mlp.fc1.bias conv2d_15node_conv2d_15"Conv* @@ -706,47 +706,47 @@ Cvision_model.vision._en.stages.1.blocks.0.mlp.conv.conv.weight_bias getitem_6 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.1: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.1.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.1.blocks.0: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.1.blocks.0.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.1.blocks.0.mlp.fc1: torch.nn.modules.conv.Conv2d/conv2d_15: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_15 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%getitem_6, %p_vision_model_vision__en_stages_1_blocks_0_mlp_fc1_weight, %p_vision_model_vision__en_stages_1_blocks_0_mlp_fc1_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.0', 'vision_model.vision._en.stages.1.blocks.0.mlp', 'vision_model.vision._en.stages.1.blocks.0.mlp.fc1', 'conv2d_15']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.0', 'vision_model.vision._en.stages.1.blocks.0.mlp', 'vision_model.vision._en.stages.1.blocks.0.mlp.fc1', 'conv2d_15']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward x = self.fc1(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_15gelu_6 node_gelu_6"Gelu* approximate"tanhJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.1: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.1.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.1.blocks.0: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.1.blocks.0.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.1.blocks.0.mlp.act: timm.layers.activations.GELUTanh/gelu_6: aten.gelu.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'timm.layers.activations.GELUTanh', 'aten.gelu.default']J pkg.torch.onnx.fx_node%gelu_6 : [num_users=1] = call_function[target=torch.ops.aten.gelu.default](args = (%conv2d_15,), kwargs = {approximate: tanh})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.0', 'vision_model.vision._en.stages.1.blocks.0.mlp', 'vision_model.vision._en.stages.1.blocks.0.mlp.act', 'gelu_6']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.0', 'vision_model.vision._en.stages.1.blocks.0.mlp', 'vision_model.vision._en.stages.1.blocks.0.mlp.act', 'gelu_6']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, in forward x = self.act(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') - + gelu_6 8vision_model.vision._en.stages.1.blocks.0.mlp.fc2.weight 6vision_model.vision._en.stages.1.blocks.0.mlp.fc2.bias conv2d_16node_conv2d_16"Conv* @@ -758,66 +758,66 @@ Cvision_model.vision._en.stages.1.blocks.0.mlp.conv.conv.weight_bias getitem_6 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.1: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.1.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.1.blocks.0: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.1.blocks.0.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.1.blocks.0.mlp.fc2: torch.nn.modules.conv.Conv2d/conv2d_16: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_16 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%clone_4, %p_vision_model_vision__en_stages_1_blocks_0_mlp_fc2_weight, %p_vision_model_vision__en_stages_1_blocks_0_mlp_fc2_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.0', 'vision_model.vision._en.stages.1.blocks.0.mlp', 'vision_model.vision._en.stages.1.blocks.0.mlp.fc2', 'conv2d_16']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.0', 'vision_model.vision._en.stages.1.blocks.0.mlp', 'vision_model.vision._en.stages.1.blocks.0.mlp.fc2', 'conv2d_16']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward x = self.fc2(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_16 ;vision_model.vision._en.stages.1.blocks.0.layer_scale.gammamul_2 node_mul_2"MulJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.1: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.1.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.1.blocks.0: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.1.blocks.0.layer_scale: timm.models.fastvit.LayerScale2d/mul_2: aten.mul.TensorJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.LayerScale2d', 'aten.mul.Tensor']J pkg.torch.onnx.fx_node%mul_2 : [num_users=1] = call_function[target=torch.ops.aten.mul.Tensor](args = (%clone_5, %p_vision_model_vision__en_stages_1_blocks_0_layer_scale_gamma), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.0', 'vision_model.vision._en.stages.1.blocks.0.layer_scale', 'mul_2']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.0', 'vision_model.vision._en.stages.1.blocks.0.layer_scale', 'mul_2']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 646, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 646, in forward return x.mul_(self.gamma) if self.inplace else x * self.gamma - + conv2d_13 mul_2add_2 node_add_2"AddJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.1: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.1.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.1.blocks.0: timm.models.fastvit.RepMixerBlock/add_2: aten.add.TensorJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'aten.add.Tensor']J pkg.torch.onnx.fx_noder%add_2 : [num_users=1] = call_function[target=torch.ops.aten.add.Tensor](args = (%conv2d_13, %mul_2), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.0', 'add_2']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.0', 'add_2']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - + add_2 Ivision_model.vision._en.stages.1.blocks.1.token_mixer.reparam_conv.weight Gvision_model.vision._en.stages.1.blocks.1.token_mixer.reparam_conv.bias conv2d_17node_conv2d_17"Conv* @@ -829,24 +829,24 @@ Gvision_model.vision._en.stages.1.blocks.1.token_mixer.reparam_conv.bias conv2d namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.1: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.1.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.1.blocks.1: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.1.blocks.1.token_mixer: timm.models.fastvit.RepMixer/vision_model.vision._en.stages.1.blocks.1.token_mixer.reparam_conv: torch.nn.modules.conv.Conv2d/conv2d_17: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.RepMixer', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_17 : [num_users=2] = call_function[target=torch.ops.aten.conv2d.default](args = (%add_2, %p_vision_model_vision__en_stages_1_blocks_1_token_mixer_reparam_conv_weight, %p_vision_model_vision__en_stages_1_blocks_1_token_mixer_reparam_conv_bias, [1, 1], [1, 1], [1, 1], 128), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.1', 'vision_model.vision._en.stages.1.blocks.1.token_mixer', 'vision_model.vision._en.stages.1.blocks.1.token_mixer.reparam_conv', 'conv2d_17']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.1', 'vision_model.vision._en.stages.1.blocks.1.token_mixer', 'vision_model.vision._en.stages.1.blocks.1.token_mixer.reparam_conv', 'conv2d_17']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1001, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1001, in forward x = self.token_mixer(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward x = self.reparam_conv(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_17 >vision_model.vision._en.stages.1.blocks.1.mlp.conv.conv.weight Cvision_model.vision._en.stages.1.blocks.1.mlp.conv.conv.weight_bias getitem_9 node_Conv_188"Conv* @@ -859,26 +859,26 @@ Cvision_model.vision._en.stages.1.blocks.1.mlp.conv.conv.weight_bias getitem_9 !pkg.onnxscript.rewriter.rule_name1FuseBatchNormIntoConv, RemoveOptionalBiasFromConvJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'timm.layers.conv_bn_act.ConvNormAct', 'timm.layers.norm_act.BatchNormAct2d', 'aten._native_batch_norm_legit_no_training.default']J pkg.torch.onnx.fx_node%_native_batch_norm_legit_no_training_3 : [num_users=1] = call_function[target=torch.ops.aten._native_batch_norm_legit_no_training.default](args = (%conv2d_18, %p_vision_model_vision__en_stages_1_blocks_1_mlp_conv_bn_weight, %p_vision_model_vision__en_stages_1_blocks_1_mlp_conv_bn_bias, %b_vision_model_vision__en_stages_1_blocks_1_mlp_conv_bn_running_mean, %b_vision_model_vision__en_stages_1_blocks_1_mlp_conv_bn_running_var, 0.1, 1e-05), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.1', 'vision_model.vision._en.stages.1.blocks.1.mlp', 'vision_model.vision._en.stages.1.blocks.1.mlp.conv', 'vision_model.vision._en.stages.1.blocks.1.mlp.conv.bn', '_native_batch_norm_legit_no_training_3']J -pkg.torch.onnx.stack_trace File "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.1', 'vision_model.vision._en.stages.1.blocks.1.mlp', 'vision_model.vision._en.stages.1.blocks.1.mlp.conv', 'vision_model.vision._en.stages.1.blocks.1.mlp.conv.bn', '_native_batch_norm_legit_no_training_3']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 814, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 814, in forward x = self.conv(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/conv_bn_act.py", line 94, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/conv_bn_act.py", line 94, in forward x = self.bn(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward x = F.batch_norm( - + getitem_9 8vision_model.vision._en.stages.1.blocks.1.mlp.fc1.weight 6vision_model.vision._en.stages.1.blocks.1.mlp.fc1.bias conv2d_19node_conv2d_19"Conv* @@ -890,47 +890,47 @@ Cvision_model.vision._en.stages.1.blocks.1.mlp.conv.conv.weight_bias getitem_9 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.1: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.1.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.1.blocks.1: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.1.blocks.1.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.1.blocks.1.mlp.fc1: torch.nn.modules.conv.Conv2d/conv2d_19: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_19 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%getitem_9, %p_vision_model_vision__en_stages_1_blocks_1_mlp_fc1_weight, %p_vision_model_vision__en_stages_1_blocks_1_mlp_fc1_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.1', 'vision_model.vision._en.stages.1.blocks.1.mlp', 'vision_model.vision._en.stages.1.blocks.1.mlp.fc1', 'conv2d_19']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.1', 'vision_model.vision._en.stages.1.blocks.1.mlp', 'vision_model.vision._en.stages.1.blocks.1.mlp.fc1', 'conv2d_19']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward x = self.fc1(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_19gelu_7 node_gelu_7"Gelu* approximate"tanhJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.1: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.1.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.1.blocks.1: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.1.blocks.1.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.1.blocks.1.mlp.act: timm.layers.activations.GELUTanh/gelu_7: aten.gelu.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'timm.layers.activations.GELUTanh', 'aten.gelu.default']J pkg.torch.onnx.fx_node%gelu_7 : [num_users=1] = call_function[target=torch.ops.aten.gelu.default](args = (%conv2d_19,), kwargs = {approximate: tanh})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.1', 'vision_model.vision._en.stages.1.blocks.1.mlp', 'vision_model.vision._en.stages.1.blocks.1.mlp.act', 'gelu_7']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.1', 'vision_model.vision._en.stages.1.blocks.1.mlp', 'vision_model.vision._en.stages.1.blocks.1.mlp.act', 'gelu_7']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, in forward x = self.act(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') - + gelu_7 8vision_model.vision._en.stages.1.blocks.1.mlp.fc2.weight 6vision_model.vision._en.stages.1.blocks.1.mlp.fc2.bias conv2d_20node_conv2d_20"Conv* @@ -942,66 +942,66 @@ Cvision_model.vision._en.stages.1.blocks.1.mlp.conv.conv.weight_bias getitem_9 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.1: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.1.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.1.blocks.1: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.1.blocks.1.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.1.blocks.1.mlp.fc2: torch.nn.modules.conv.Conv2d/conv2d_20: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_20 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%clone_6, %p_vision_model_vision__en_stages_1_blocks_1_mlp_fc2_weight, %p_vision_model_vision__en_stages_1_blocks_1_mlp_fc2_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.1', 'vision_model.vision._en.stages.1.blocks.1.mlp', 'vision_model.vision._en.stages.1.blocks.1.mlp.fc2', 'conv2d_20']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.1', 'vision_model.vision._en.stages.1.blocks.1.mlp', 'vision_model.vision._en.stages.1.blocks.1.mlp.fc2', 'conv2d_20']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward x = self.fc2(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_20 ;vision_model.vision._en.stages.1.blocks.1.layer_scale.gammamul_3 node_mul_3"MulJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.1: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.1.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.1.blocks.1: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.1.blocks.1.layer_scale: timm.models.fastvit.LayerScale2d/mul_3: aten.mul.TensorJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.LayerScale2d', 'aten.mul.Tensor']J pkg.torch.onnx.fx_node%mul_3 : [num_users=1] = call_function[target=torch.ops.aten.mul.Tensor](args = (%clone_7, %p_vision_model_vision__en_stages_1_blocks_1_layer_scale_gamma), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.1', 'vision_model.vision._en.stages.1.blocks.1.layer_scale', 'mul_3']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.1', 'vision_model.vision._en.stages.1.blocks.1.layer_scale', 'mul_3']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 646, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 646, in forward return x.mul_(self.gamma) if self.inplace else x * self.gamma - + conv2d_17 mul_3add_3 node_add_3"AddJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.1: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.1.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.1.blocks.1: timm.models.fastvit.RepMixerBlock/add_3: aten.add.TensorJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'aten.add.Tensor']J pkg.torch.onnx.fx_noder%add_3 : [num_users=1] = call_function[target=torch.ops.aten.add.Tensor](args = (%conv2d_17, %mul_3), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.1', 'add_3']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.1', 'vision_model.vision._en.stages.1.blocks', 'vision_model.vision._en.stages.1.blocks.1', 'add_3']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - + add_3 Fvision_model.vision._en.stages.2.downsample.proj.0.reparam_conv.weight Dvision_model.vision._en.stages.2.downsample.proj.0.reparam_conv.bias conv2d_21node_conv2d_21"Conv* @@ -1013,24 +1013,24 @@ Dvision_model.vision._en.stages.2.downsample.proj.0.reparam_conv.bias conv2d_21 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.downsample: timm.models.fastvit.PatchEmbed/vision_model.vision._en.stages.2.downsample.proj: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.downsample.proj.0: timm.models.fastvit.ReparamLargeKernelConv/vision_model.vision._en.stages.2.downsample.proj.0.reparam_conv: torch.nn.modules.conv.Conv2d/conv2d_21: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'timm.models.fastvit.PatchEmbed', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.ReparamLargeKernelConv', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_21 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%add_3, %p_vision_model_vision__en_stages_2_downsample_proj_0_reparam_conv_weight, %p_vision_model_vision__en_stages_2_downsample_proj_0_reparam_conv_bias, [2, 2], [3, 3], [1, 1], 128), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.downsample', 'vision_model.vision._en.stages.2.downsample.proj', 'vision_model.vision._en.stages.2.downsample.proj.0', 'vision_model.vision._en.stages.2.downsample.proj.0.reparam_conv', 'conv2d_21']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.downsample', 'vision_model.vision._en.stages.2.downsample.proj', 'vision_model.vision._en.stages.2.downsample.proj.0', 'vision_model.vision._en.stages.2.downsample.proj.0.reparam_conv', 'conv2d_21']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1162, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1162, in forward x = self.downsample(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 628, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 628, in forward x = self.proj(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 376, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 376, in forward out = self.reparam_conv(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_21 Fvision_model.vision._en.stages.2.downsample.proj.1.reparam_conv.weight Dvision_model.vision._en.stages.2.downsample.proj.1.reparam_conv.bias conv2d_22node_conv2d_22"Conv* @@ -1042,47 +1042,47 @@ Dvision_model.vision._en.stages.2.downsample.proj.1.reparam_conv.bias conv2d_22 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.downsample: timm.models.fastvit.PatchEmbed/vision_model.vision._en.stages.2.downsample.proj: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.downsample.proj.1: timm.models.fastvit.MobileOneBlock/vision_model.vision._en.stages.2.downsample.proj.1.reparam_conv: torch.nn.modules.conv.Conv2d/conv2d_22: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'timm.models.fastvit.PatchEmbed', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.MobileOneBlock', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_22 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%conv2d_21, %p_vision_model_vision__en_stages_2_downsample_proj_1_reparam_conv_weight, %p_vision_model_vision__en_stages_2_downsample_proj_1_reparam_conv_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.downsample', 'vision_model.vision._en.stages.2.downsample.proj', 'vision_model.vision._en.stages.2.downsample.proj.1', 'vision_model.vision._en.stages.2.downsample.proj.1.reparam_conv', 'conv2d_22']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.downsample', 'vision_model.vision._en.stages.2.downsample.proj', 'vision_model.vision._en.stages.2.downsample.proj.1', 'vision_model.vision._en.stages.2.downsample.proj.1.reparam_conv', 'conv2d_22']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1162, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1162, in forward x = self.downsample(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 628, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 628, in forward x = self.proj(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward return self.act(self.se(self.reparam_conv(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_22gelu_8 node_gelu_8"Gelu* approximate"tanhJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.downsample: timm.models.fastvit.PatchEmbed/vision_model.vision._en.stages.2.downsample.proj: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.downsample.proj.1: timm.models.fastvit.MobileOneBlock/vision_model.vision._en.stages.2.downsample.proj.1.act: timm.layers.activations.GELUTanh/gelu_8: aten.gelu.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'timm.models.fastvit.PatchEmbed', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.MobileOneBlock', 'timm.layers.activations.GELUTanh', 'aten.gelu.default']J pkg.torch.onnx.fx_node%gelu_8 : [num_users=1] = call_function[target=torch.ops.aten.gelu.default](args = (%conv2d_22,), kwargs = {approximate: tanh})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.downsample', 'vision_model.vision._en.stages.2.downsample.proj', 'vision_model.vision._en.stages.2.downsample.proj.1', 'vision_model.vision._en.stages.2.downsample.proj.1.act', 'gelu_8']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.downsample', 'vision_model.vision._en.stages.2.downsample.proj', 'vision_model.vision._en.stages.2.downsample.proj.1', 'vision_model.vision._en.stages.2.downsample.proj.1.act', 'gelu_8']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1162, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1162, in forward x = self.downsample(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 628, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 628, in forward x = self.proj(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward return self.act(self.se(self.reparam_conv(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') - + gelu_8 Ivision_model.vision._en.stages.2.blocks.0.token_mixer.reparam_conv.weight Gvision_model.vision._en.stages.2.blocks.0.token_mixer.reparam_conv.bias conv2d_23node_conv2d_23"Conv* @@ -1094,24 +1094,24 @@ Gvision_model.vision._en.stages.2.blocks.0.token_mixer.reparam_conv.bias conv2d namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.0: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.0.token_mixer: timm.models.fastvit.RepMixer/vision_model.vision._en.stages.2.blocks.0.token_mixer.reparam_conv: torch.nn.modules.conv.Conv2d/conv2d_23: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.RepMixer', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_23 : [num_users=2] = call_function[target=torch.ops.aten.conv2d.default](args = (%gelu_8, %p_vision_model_vision__en_stages_2_blocks_0_token_mixer_reparam_conv_weight, %p_vision_model_vision__en_stages_2_blocks_0_token_mixer_reparam_conv_bias, [1, 1], [1, 1], [1, 1], 256), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.0', 'vision_model.vision._en.stages.2.blocks.0.token_mixer', 'vision_model.vision._en.stages.2.blocks.0.token_mixer.reparam_conv', 'conv2d_23']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.0', 'vision_model.vision._en.stages.2.blocks.0.token_mixer', 'vision_model.vision._en.stages.2.blocks.0.token_mixer.reparam_conv', 'conv2d_23']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1001, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1001, in forward x = self.token_mixer(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward x = self.reparam_conv(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_23 >vision_model.vision._en.stages.2.blocks.0.mlp.conv.conv.weight Cvision_model.vision._en.stages.2.blocks.0.mlp.conv.conv.weight_bias @@ -1125,26 +1125,26 @@ getitem_12 node_Conv_190"Conv* !pkg.onnxscript.rewriter.rule_name1FuseBatchNormIntoConv, RemoveOptionalBiasFromConvJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'timm.layers.conv_bn_act.ConvNormAct', 'timm.layers.norm_act.BatchNormAct2d', 'aten._native_batch_norm_legit_no_training.default']J pkg.torch.onnx.fx_node%_native_batch_norm_legit_no_training_4 : [num_users=1] = call_function[target=torch.ops.aten._native_batch_norm_legit_no_training.default](args = (%conv2d_24, %p_vision_model_vision__en_stages_2_blocks_0_mlp_conv_bn_weight, %p_vision_model_vision__en_stages_2_blocks_0_mlp_conv_bn_bias, %b_vision_model_vision__en_stages_2_blocks_0_mlp_conv_bn_running_mean, %b_vision_model_vision__en_stages_2_blocks_0_mlp_conv_bn_running_var, 0.1, 1e-05), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.0', 'vision_model.vision._en.stages.2.blocks.0.mlp', 'vision_model.vision._en.stages.2.blocks.0.mlp.conv', 'vision_model.vision._en.stages.2.blocks.0.mlp.conv.bn', '_native_batch_norm_legit_no_training_4']J -pkg.torch.onnx.stack_trace File "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.0', 'vision_model.vision._en.stages.2.blocks.0.mlp', 'vision_model.vision._en.stages.2.blocks.0.mlp.conv', 'vision_model.vision._en.stages.2.blocks.0.mlp.conv.bn', '_native_batch_norm_legit_no_training_4']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 814, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 814, in forward x = self.conv(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/conv_bn_act.py", line 94, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/conv_bn_act.py", line 94, in forward x = self.bn(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward x = F.batch_norm( - + getitem_12 8vision_model.vision._en.stages.2.blocks.0.mlp.fc1.weight @@ -1157,47 +1157,47 @@ getitem_12 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.0: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.0.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.2.blocks.0.mlp.fc1: torch.nn.modules.conv.Conv2d/conv2d_25: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_25 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%getitem_12, %p_vision_model_vision__en_stages_2_blocks_0_mlp_fc1_weight, %p_vision_model_vision__en_stages_2_blocks_0_mlp_fc1_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.0', 'vision_model.vision._en.stages.2.blocks.0.mlp', 'vision_model.vision._en.stages.2.blocks.0.mlp.fc1', 'conv2d_25']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.0', 'vision_model.vision._en.stages.2.blocks.0.mlp', 'vision_model.vision._en.stages.2.blocks.0.mlp.fc1', 'conv2d_25']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward x = self.fc1(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_25gelu_9 node_gelu_9"Gelu* approximate"tanhJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.0: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.0.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.2.blocks.0.mlp.act: timm.layers.activations.GELUTanh/gelu_9: aten.gelu.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'timm.layers.activations.GELUTanh', 'aten.gelu.default']J pkg.torch.onnx.fx_node%gelu_9 : [num_users=1] = call_function[target=torch.ops.aten.gelu.default](args = (%conv2d_25,), kwargs = {approximate: tanh})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.0', 'vision_model.vision._en.stages.2.blocks.0.mlp', 'vision_model.vision._en.stages.2.blocks.0.mlp.act', 'gelu_9']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.0', 'vision_model.vision._en.stages.2.blocks.0.mlp', 'vision_model.vision._en.stages.2.blocks.0.mlp.act', 'gelu_9']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, in forward x = self.act(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') - + gelu_9 8vision_model.vision._en.stages.2.blocks.0.mlp.fc2.weight 6vision_model.vision._en.stages.2.blocks.0.mlp.fc2.bias conv2d_26node_conv2d_26"Conv* @@ -1209,66 +1209,66 @@ getitem_12 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.0: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.0.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.2.blocks.0.mlp.fc2: torch.nn.modules.conv.Conv2d/conv2d_26: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_26 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%clone_8, %p_vision_model_vision__en_stages_2_blocks_0_mlp_fc2_weight, %p_vision_model_vision__en_stages_2_blocks_0_mlp_fc2_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.0', 'vision_model.vision._en.stages.2.blocks.0.mlp', 'vision_model.vision._en.stages.2.blocks.0.mlp.fc2', 'conv2d_26']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.0', 'vision_model.vision._en.stages.2.blocks.0.mlp', 'vision_model.vision._en.stages.2.blocks.0.mlp.fc2', 'conv2d_26']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward x = self.fc2(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_26 ;vision_model.vision._en.stages.2.blocks.0.layer_scale.gammamul_4 node_mul_4"MulJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.0: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.0.layer_scale: timm.models.fastvit.LayerScale2d/mul_4: aten.mul.TensorJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.LayerScale2d', 'aten.mul.Tensor']J pkg.torch.onnx.fx_node%mul_4 : [num_users=1] = call_function[target=torch.ops.aten.mul.Tensor](args = (%clone_9, %p_vision_model_vision__en_stages_2_blocks_0_layer_scale_gamma), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.0', 'vision_model.vision._en.stages.2.blocks.0.layer_scale', 'mul_4']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.0', 'vision_model.vision._en.stages.2.blocks.0.layer_scale', 'mul_4']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 646, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 646, in forward return x.mul_(self.gamma) if self.inplace else x * self.gamma - + conv2d_23 mul_4add_4 node_add_4"AddJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.0: timm.models.fastvit.RepMixerBlock/add_4: aten.add.TensorJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'aten.add.Tensor']J pkg.torch.onnx.fx_noder%add_4 : [num_users=1] = call_function[target=torch.ops.aten.add.Tensor](args = (%conv2d_23, %mul_4), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.0', 'add_4']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.0', 'add_4']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - + add_4 Ivision_model.vision._en.stages.2.blocks.1.token_mixer.reparam_conv.weight Gvision_model.vision._en.stages.2.blocks.1.token_mixer.reparam_conv.bias conv2d_27node_conv2d_27"Conv* @@ -1280,24 +1280,24 @@ Gvision_model.vision._en.stages.2.blocks.1.token_mixer.reparam_conv.bias conv2d namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.1: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.1.token_mixer: timm.models.fastvit.RepMixer/vision_model.vision._en.stages.2.blocks.1.token_mixer.reparam_conv: torch.nn.modules.conv.Conv2d/conv2d_27: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.RepMixer', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_27 : [num_users=2] = call_function[target=torch.ops.aten.conv2d.default](args = (%add_4, %p_vision_model_vision__en_stages_2_blocks_1_token_mixer_reparam_conv_weight, %p_vision_model_vision__en_stages_2_blocks_1_token_mixer_reparam_conv_bias, [1, 1], [1, 1], [1, 1], 256), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.1', 'vision_model.vision._en.stages.2.blocks.1.token_mixer', 'vision_model.vision._en.stages.2.blocks.1.token_mixer.reparam_conv', 'conv2d_27']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.1', 'vision_model.vision._en.stages.2.blocks.1.token_mixer', 'vision_model.vision._en.stages.2.blocks.1.token_mixer.reparam_conv', 'conv2d_27']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1001, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1001, in forward x = self.token_mixer(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward x = self.reparam_conv(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_27 >vision_model.vision._en.stages.2.blocks.1.mlp.conv.conv.weight Cvision_model.vision._en.stages.2.blocks.1.mlp.conv.conv.weight_bias @@ -1311,26 +1311,26 @@ getitem_15 node_Conv_192"Conv* !pkg.onnxscript.rewriter.rule_name1FuseBatchNormIntoConv, RemoveOptionalBiasFromConvJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'timm.layers.conv_bn_act.ConvNormAct', 'timm.layers.norm_act.BatchNormAct2d', 'aten._native_batch_norm_legit_no_training.default']J pkg.torch.onnx.fx_node%_native_batch_norm_legit_no_training_5 : [num_users=1] = call_function[target=torch.ops.aten._native_batch_norm_legit_no_training.default](args = (%conv2d_28, %p_vision_model_vision__en_stages_2_blocks_1_mlp_conv_bn_weight, %p_vision_model_vision__en_stages_2_blocks_1_mlp_conv_bn_bias, %b_vision_model_vision__en_stages_2_blocks_1_mlp_conv_bn_running_mean, %b_vision_model_vision__en_stages_2_blocks_1_mlp_conv_bn_running_var, 0.1, 1e-05), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.1', 'vision_model.vision._en.stages.2.blocks.1.mlp', 'vision_model.vision._en.stages.2.blocks.1.mlp.conv', 'vision_model.vision._en.stages.2.blocks.1.mlp.conv.bn', '_native_batch_norm_legit_no_training_5']J -pkg.torch.onnx.stack_trace File "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.1', 'vision_model.vision._en.stages.2.blocks.1.mlp', 'vision_model.vision._en.stages.2.blocks.1.mlp.conv', 'vision_model.vision._en.stages.2.blocks.1.mlp.conv.bn', '_native_batch_norm_legit_no_training_5']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 814, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 814, in forward x = self.conv(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/conv_bn_act.py", line 94, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/conv_bn_act.py", line 94, in forward x = self.bn(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward x = F.batch_norm( - + getitem_15 8vision_model.vision._en.stages.2.blocks.1.mlp.fc1.weight @@ -1343,47 +1343,47 @@ getitem_15 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.1: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.1.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.2.blocks.1.mlp.fc1: torch.nn.modules.conv.Conv2d/conv2d_29: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_29 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%getitem_15, %p_vision_model_vision__en_stages_2_blocks_1_mlp_fc1_weight, %p_vision_model_vision__en_stages_2_blocks_1_mlp_fc1_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.1', 'vision_model.vision._en.stages.2.blocks.1.mlp', 'vision_model.vision._en.stages.2.blocks.1.mlp.fc1', 'conv2d_29']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.1', 'vision_model.vision._en.stages.2.blocks.1.mlp', 'vision_model.vision._en.stages.2.blocks.1.mlp.fc1', 'conv2d_29']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward x = self.fc1(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_29gelu_10 node_gelu_10"Gelu* approximate"tanhJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.1: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.1.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.2.blocks.1.mlp.act: timm.layers.activations.GELUTanh/gelu_10: aten.gelu.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'timm.layers.activations.GELUTanh', 'aten.gelu.default']J pkg.torch.onnx.fx_node%gelu_10 : [num_users=1] = call_function[target=torch.ops.aten.gelu.default](args = (%conv2d_29,), kwargs = {approximate: tanh})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.1', 'vision_model.vision._en.stages.2.blocks.1.mlp', 'vision_model.vision._en.stages.2.blocks.1.mlp.act', 'gelu_10']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.1', 'vision_model.vision._en.stages.2.blocks.1.mlp', 'vision_model.vision._en.stages.2.blocks.1.mlp.act', 'gelu_10']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, in forward x = self.act(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') - + gelu_10 8vision_model.vision._en.stages.2.blocks.1.mlp.fc2.weight 6vision_model.vision._en.stages.2.blocks.1.mlp.fc2.bias conv2d_30node_conv2d_30"Conv* @@ -1395,66 +1395,66 @@ getitem_15 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.1: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.1.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.2.blocks.1.mlp.fc2: torch.nn.modules.conv.Conv2d/conv2d_30: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_30 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%clone_10, %p_vision_model_vision__en_stages_2_blocks_1_mlp_fc2_weight, %p_vision_model_vision__en_stages_2_blocks_1_mlp_fc2_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.1', 'vision_model.vision._en.stages.2.blocks.1.mlp', 'vision_model.vision._en.stages.2.blocks.1.mlp.fc2', 'conv2d_30']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.1', 'vision_model.vision._en.stages.2.blocks.1.mlp', 'vision_model.vision._en.stages.2.blocks.1.mlp.fc2', 'conv2d_30']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward x = self.fc2(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_30 ;vision_model.vision._en.stages.2.blocks.1.layer_scale.gammamul_5 node_mul_5"MulJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.1: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.1.layer_scale: timm.models.fastvit.LayerScale2d/mul_5: aten.mul.TensorJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.LayerScale2d', 'aten.mul.Tensor']J pkg.torch.onnx.fx_node%mul_5 : [num_users=1] = call_function[target=torch.ops.aten.mul.Tensor](args = (%clone_11, %p_vision_model_vision__en_stages_2_blocks_1_layer_scale_gamma), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.1', 'vision_model.vision._en.stages.2.blocks.1.layer_scale', 'mul_5']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.1', 'vision_model.vision._en.stages.2.blocks.1.layer_scale', 'mul_5']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 646, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 646, in forward return x.mul_(self.gamma) if self.inplace else x * self.gamma - + conv2d_27 mul_5add_5 node_add_5"AddJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.1: timm.models.fastvit.RepMixerBlock/add_5: aten.add.TensorJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'aten.add.Tensor']J pkg.torch.onnx.fx_noder%add_5 : [num_users=1] = call_function[target=torch.ops.aten.add.Tensor](args = (%conv2d_27, %mul_5), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.1', 'add_5']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.1', 'add_5']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - + add_5 Ivision_model.vision._en.stages.2.blocks.2.token_mixer.reparam_conv.weight Gvision_model.vision._en.stages.2.blocks.2.token_mixer.reparam_conv.bias conv2d_31node_conv2d_31"Conv* @@ -1466,24 +1466,24 @@ Gvision_model.vision._en.stages.2.blocks.2.token_mixer.reparam_conv.bias conv2d namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.2: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.2.token_mixer: timm.models.fastvit.RepMixer/vision_model.vision._en.stages.2.blocks.2.token_mixer.reparam_conv: torch.nn.modules.conv.Conv2d/conv2d_31: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.RepMixer', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_31 : [num_users=2] = call_function[target=torch.ops.aten.conv2d.default](args = (%add_5, %p_vision_model_vision__en_stages_2_blocks_2_token_mixer_reparam_conv_weight, %p_vision_model_vision__en_stages_2_blocks_2_token_mixer_reparam_conv_bias, [1, 1], [1, 1], [1, 1], 256), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.2', 'vision_model.vision._en.stages.2.blocks.2.token_mixer', 'vision_model.vision._en.stages.2.blocks.2.token_mixer.reparam_conv', 'conv2d_31']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.2', 'vision_model.vision._en.stages.2.blocks.2.token_mixer', 'vision_model.vision._en.stages.2.blocks.2.token_mixer.reparam_conv', 'conv2d_31']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1001, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1001, in forward x = self.token_mixer(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward x = self.reparam_conv(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_31 >vision_model.vision._en.stages.2.blocks.2.mlp.conv.conv.weight Cvision_model.vision._en.stages.2.blocks.2.mlp.conv.conv.weight_bias @@ -1497,26 +1497,26 @@ getitem_18 node_Conv_194"Conv* !pkg.onnxscript.rewriter.rule_name1FuseBatchNormIntoConv, RemoveOptionalBiasFromConvJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'timm.layers.conv_bn_act.ConvNormAct', 'timm.layers.norm_act.BatchNormAct2d', 'aten._native_batch_norm_legit_no_training.default']J pkg.torch.onnx.fx_node%_native_batch_norm_legit_no_training_6 : [num_users=1] = call_function[target=torch.ops.aten._native_batch_norm_legit_no_training.default](args = (%conv2d_32, %p_vision_model_vision__en_stages_2_blocks_2_mlp_conv_bn_weight, %p_vision_model_vision__en_stages_2_blocks_2_mlp_conv_bn_bias, %b_vision_model_vision__en_stages_2_blocks_2_mlp_conv_bn_running_mean, %b_vision_model_vision__en_stages_2_blocks_2_mlp_conv_bn_running_var, 0.1, 1e-05), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.2', 'vision_model.vision._en.stages.2.blocks.2.mlp', 'vision_model.vision._en.stages.2.blocks.2.mlp.conv', 'vision_model.vision._en.stages.2.blocks.2.mlp.conv.bn', '_native_batch_norm_legit_no_training_6']J -pkg.torch.onnx.stack_trace File "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.2', 'vision_model.vision._en.stages.2.blocks.2.mlp', 'vision_model.vision._en.stages.2.blocks.2.mlp.conv', 'vision_model.vision._en.stages.2.blocks.2.mlp.conv.bn', '_native_batch_norm_legit_no_training_6']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 814, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 814, in forward x = self.conv(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/conv_bn_act.py", line 94, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/conv_bn_act.py", line 94, in forward x = self.bn(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward x = F.batch_norm( - + getitem_18 8vision_model.vision._en.stages.2.blocks.2.mlp.fc1.weight @@ -1529,47 +1529,47 @@ getitem_18 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.2: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.2.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.2.blocks.2.mlp.fc1: torch.nn.modules.conv.Conv2d/conv2d_33: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_33 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%getitem_18, %p_vision_model_vision__en_stages_2_blocks_2_mlp_fc1_weight, %p_vision_model_vision__en_stages_2_blocks_2_mlp_fc1_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.2', 'vision_model.vision._en.stages.2.blocks.2.mlp', 'vision_model.vision._en.stages.2.blocks.2.mlp.fc1', 'conv2d_33']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.2', 'vision_model.vision._en.stages.2.blocks.2.mlp', 'vision_model.vision._en.stages.2.blocks.2.mlp.fc1', 'conv2d_33']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward x = self.fc1(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_33gelu_11 node_gelu_11"Gelu* approximate"tanhJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.2: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.2.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.2.blocks.2.mlp.act: timm.layers.activations.GELUTanh/gelu_11: aten.gelu.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'timm.layers.activations.GELUTanh', 'aten.gelu.default']J pkg.torch.onnx.fx_node%gelu_11 : [num_users=1] = call_function[target=torch.ops.aten.gelu.default](args = (%conv2d_33,), kwargs = {approximate: tanh})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.2', 'vision_model.vision._en.stages.2.blocks.2.mlp', 'vision_model.vision._en.stages.2.blocks.2.mlp.act', 'gelu_11']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.2', 'vision_model.vision._en.stages.2.blocks.2.mlp', 'vision_model.vision._en.stages.2.blocks.2.mlp.act', 'gelu_11']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, in forward x = self.act(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') - + gelu_11 8vision_model.vision._en.stages.2.blocks.2.mlp.fc2.weight 6vision_model.vision._en.stages.2.blocks.2.mlp.fc2.bias conv2d_34node_conv2d_34"Conv* @@ -1581,66 +1581,66 @@ getitem_18 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.2: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.2.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.2.blocks.2.mlp.fc2: torch.nn.modules.conv.Conv2d/conv2d_34: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_34 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%clone_12, %p_vision_model_vision__en_stages_2_blocks_2_mlp_fc2_weight, %p_vision_model_vision__en_stages_2_blocks_2_mlp_fc2_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.2', 'vision_model.vision._en.stages.2.blocks.2.mlp', 'vision_model.vision._en.stages.2.blocks.2.mlp.fc2', 'conv2d_34']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.2', 'vision_model.vision._en.stages.2.blocks.2.mlp', 'vision_model.vision._en.stages.2.blocks.2.mlp.fc2', 'conv2d_34']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward x = self.fc2(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_34 ;vision_model.vision._en.stages.2.blocks.2.layer_scale.gammamul_6 node_mul_6"MulJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.2: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.2.layer_scale: timm.models.fastvit.LayerScale2d/mul_6: aten.mul.TensorJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.LayerScale2d', 'aten.mul.Tensor']J pkg.torch.onnx.fx_node%mul_6 : [num_users=1] = call_function[target=torch.ops.aten.mul.Tensor](args = (%clone_13, %p_vision_model_vision__en_stages_2_blocks_2_layer_scale_gamma), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.2', 'vision_model.vision._en.stages.2.blocks.2.layer_scale', 'mul_6']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.2', 'vision_model.vision._en.stages.2.blocks.2.layer_scale', 'mul_6']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 646, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 646, in forward return x.mul_(self.gamma) if self.inplace else x * self.gamma - + conv2d_31 mul_6add_6 node_add_6"AddJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.2: timm.models.fastvit.RepMixerBlock/add_6: aten.add.TensorJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'aten.add.Tensor']J pkg.torch.onnx.fx_noder%add_6 : [num_users=1] = call_function[target=torch.ops.aten.add.Tensor](args = (%conv2d_31, %mul_6), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.2', 'add_6']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.2', 'add_6']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - + add_6 Ivision_model.vision._en.stages.2.blocks.3.token_mixer.reparam_conv.weight Gvision_model.vision._en.stages.2.blocks.3.token_mixer.reparam_conv.bias conv2d_35node_conv2d_35"Conv* @@ -1652,24 +1652,24 @@ Gvision_model.vision._en.stages.2.blocks.3.token_mixer.reparam_conv.bias conv2d namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.3: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.3.token_mixer: timm.models.fastvit.RepMixer/vision_model.vision._en.stages.2.blocks.3.token_mixer.reparam_conv: torch.nn.modules.conv.Conv2d/conv2d_35: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.RepMixer', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_35 : [num_users=2] = call_function[target=torch.ops.aten.conv2d.default](args = (%add_6, %p_vision_model_vision__en_stages_2_blocks_3_token_mixer_reparam_conv_weight, %p_vision_model_vision__en_stages_2_blocks_3_token_mixer_reparam_conv_bias, [1, 1], [1, 1], [1, 1], 256), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.3', 'vision_model.vision._en.stages.2.blocks.3.token_mixer', 'vision_model.vision._en.stages.2.blocks.3.token_mixer.reparam_conv', 'conv2d_35']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.3', 'vision_model.vision._en.stages.2.blocks.3.token_mixer', 'vision_model.vision._en.stages.2.blocks.3.token_mixer.reparam_conv', 'conv2d_35']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1001, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1001, in forward x = self.token_mixer(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward x = self.reparam_conv(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_35 >vision_model.vision._en.stages.2.blocks.3.mlp.conv.conv.weight Cvision_model.vision._en.stages.2.blocks.3.mlp.conv.conv.weight_bias @@ -1683,26 +1683,26 @@ getitem_21 node_Conv_196"Conv* !pkg.onnxscript.rewriter.rule_name1FuseBatchNormIntoConv, RemoveOptionalBiasFromConvJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'timm.layers.conv_bn_act.ConvNormAct', 'timm.layers.norm_act.BatchNormAct2d', 'aten._native_batch_norm_legit_no_training.default']J pkg.torch.onnx.fx_node%_native_batch_norm_legit_no_training_7 : [num_users=1] = call_function[target=torch.ops.aten._native_batch_norm_legit_no_training.default](args = (%conv2d_36, %p_vision_model_vision__en_stages_2_blocks_3_mlp_conv_bn_weight, %p_vision_model_vision__en_stages_2_blocks_3_mlp_conv_bn_bias, %b_vision_model_vision__en_stages_2_blocks_3_mlp_conv_bn_running_mean, %b_vision_model_vision__en_stages_2_blocks_3_mlp_conv_bn_running_var, 0.1, 1e-05), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.3', 'vision_model.vision._en.stages.2.blocks.3.mlp', 'vision_model.vision._en.stages.2.blocks.3.mlp.conv', 'vision_model.vision._en.stages.2.blocks.3.mlp.conv.bn', '_native_batch_norm_legit_no_training_7']J -pkg.torch.onnx.stack_trace File "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.3', 'vision_model.vision._en.stages.2.blocks.3.mlp', 'vision_model.vision._en.stages.2.blocks.3.mlp.conv', 'vision_model.vision._en.stages.2.blocks.3.mlp.conv.bn', '_native_batch_norm_legit_no_training_7']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 814, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 814, in forward x = self.conv(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/conv_bn_act.py", line 94, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/conv_bn_act.py", line 94, in forward x = self.bn(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward x = F.batch_norm( - + getitem_21 8vision_model.vision._en.stages.2.blocks.3.mlp.fc1.weight @@ -1715,47 +1715,47 @@ getitem_21 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.3: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.3.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.2.blocks.3.mlp.fc1: torch.nn.modules.conv.Conv2d/conv2d_37: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_37 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%getitem_21, %p_vision_model_vision__en_stages_2_blocks_3_mlp_fc1_weight, %p_vision_model_vision__en_stages_2_blocks_3_mlp_fc1_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.3', 'vision_model.vision._en.stages.2.blocks.3.mlp', 'vision_model.vision._en.stages.2.blocks.3.mlp.fc1', 'conv2d_37']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.3', 'vision_model.vision._en.stages.2.blocks.3.mlp', 'vision_model.vision._en.stages.2.blocks.3.mlp.fc1', 'conv2d_37']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward x = self.fc1(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_37gelu_12 node_gelu_12"Gelu* approximate"tanhJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.3: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.3.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.2.blocks.3.mlp.act: timm.layers.activations.GELUTanh/gelu_12: aten.gelu.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'timm.layers.activations.GELUTanh', 'aten.gelu.default']J pkg.torch.onnx.fx_node%gelu_12 : [num_users=1] = call_function[target=torch.ops.aten.gelu.default](args = (%conv2d_37,), kwargs = {approximate: tanh})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.3', 'vision_model.vision._en.stages.2.blocks.3.mlp', 'vision_model.vision._en.stages.2.blocks.3.mlp.act', 'gelu_12']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.3', 'vision_model.vision._en.stages.2.blocks.3.mlp', 'vision_model.vision._en.stages.2.blocks.3.mlp.act', 'gelu_12']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, in forward x = self.act(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') - + gelu_12 8vision_model.vision._en.stages.2.blocks.3.mlp.fc2.weight 6vision_model.vision._en.stages.2.blocks.3.mlp.fc2.bias conv2d_38node_conv2d_38"Conv* @@ -1767,66 +1767,66 @@ getitem_21 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.3: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.3.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.2.blocks.3.mlp.fc2: torch.nn.modules.conv.Conv2d/conv2d_38: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_38 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%clone_14, %p_vision_model_vision__en_stages_2_blocks_3_mlp_fc2_weight, %p_vision_model_vision__en_stages_2_blocks_3_mlp_fc2_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.3', 'vision_model.vision._en.stages.2.blocks.3.mlp', 'vision_model.vision._en.stages.2.blocks.3.mlp.fc2', 'conv2d_38']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.3', 'vision_model.vision._en.stages.2.blocks.3.mlp', 'vision_model.vision._en.stages.2.blocks.3.mlp.fc2', 'conv2d_38']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward x = self.fc2(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_38 ;vision_model.vision._en.stages.2.blocks.3.layer_scale.gammamul_7 node_mul_7"MulJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.3: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.3.layer_scale: timm.models.fastvit.LayerScale2d/mul_7: aten.mul.TensorJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.LayerScale2d', 'aten.mul.Tensor']J pkg.torch.onnx.fx_node%mul_7 : [num_users=1] = call_function[target=torch.ops.aten.mul.Tensor](args = (%clone_15, %p_vision_model_vision__en_stages_2_blocks_3_layer_scale_gamma), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.3', 'vision_model.vision._en.stages.2.blocks.3.layer_scale', 'mul_7']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.3', 'vision_model.vision._en.stages.2.blocks.3.layer_scale', 'mul_7']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 646, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 646, in forward return x.mul_(self.gamma) if self.inplace else x * self.gamma - + conv2d_35 mul_7add_7 node_add_7"AddJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.3: timm.models.fastvit.RepMixerBlock/add_7: aten.add.TensorJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'aten.add.Tensor']J pkg.torch.onnx.fx_noder%add_7 : [num_users=1] = call_function[target=torch.ops.aten.add.Tensor](args = (%conv2d_35, %mul_7), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.3', 'add_7']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.3', 'add_7']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - + add_7 Ivision_model.vision._en.stages.2.blocks.4.token_mixer.reparam_conv.weight Gvision_model.vision._en.stages.2.blocks.4.token_mixer.reparam_conv.bias conv2d_39node_conv2d_39"Conv* @@ -1838,24 +1838,24 @@ Gvision_model.vision._en.stages.2.blocks.4.token_mixer.reparam_conv.bias conv2d namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.4: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.4.token_mixer: timm.models.fastvit.RepMixer/vision_model.vision._en.stages.2.blocks.4.token_mixer.reparam_conv: torch.nn.modules.conv.Conv2d/conv2d_39: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.RepMixer', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_39 : [num_users=2] = call_function[target=torch.ops.aten.conv2d.default](args = (%add_7, %p_vision_model_vision__en_stages_2_blocks_4_token_mixer_reparam_conv_weight, %p_vision_model_vision__en_stages_2_blocks_4_token_mixer_reparam_conv_bias, [1, 1], [1, 1], [1, 1], 256), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.4', 'vision_model.vision._en.stages.2.blocks.4.token_mixer', 'vision_model.vision._en.stages.2.blocks.4.token_mixer.reparam_conv', 'conv2d_39']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.4', 'vision_model.vision._en.stages.2.blocks.4.token_mixer', 'vision_model.vision._en.stages.2.blocks.4.token_mixer.reparam_conv', 'conv2d_39']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1001, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1001, in forward x = self.token_mixer(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward x = self.reparam_conv(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_39 >vision_model.vision._en.stages.2.blocks.4.mlp.conv.conv.weight Cvision_model.vision._en.stages.2.blocks.4.mlp.conv.conv.weight_bias @@ -1869,26 +1869,26 @@ getitem_24 node_Conv_198"Conv* !pkg.onnxscript.rewriter.rule_name1FuseBatchNormIntoConv, RemoveOptionalBiasFromConvJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'timm.layers.conv_bn_act.ConvNormAct', 'timm.layers.norm_act.BatchNormAct2d', 'aten._native_batch_norm_legit_no_training.default']J pkg.torch.onnx.fx_node%_native_batch_norm_legit_no_training_8 : [num_users=1] = call_function[target=torch.ops.aten._native_batch_norm_legit_no_training.default](args = (%conv2d_40, %p_vision_model_vision__en_stages_2_blocks_4_mlp_conv_bn_weight, %p_vision_model_vision__en_stages_2_blocks_4_mlp_conv_bn_bias, %b_vision_model_vision__en_stages_2_blocks_4_mlp_conv_bn_running_mean, %b_vision_model_vision__en_stages_2_blocks_4_mlp_conv_bn_running_var, 0.1, 1e-05), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.4', 'vision_model.vision._en.stages.2.blocks.4.mlp', 'vision_model.vision._en.stages.2.blocks.4.mlp.conv', 'vision_model.vision._en.stages.2.blocks.4.mlp.conv.bn', '_native_batch_norm_legit_no_training_8']J -pkg.torch.onnx.stack_trace File "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.4', 'vision_model.vision._en.stages.2.blocks.4.mlp', 'vision_model.vision._en.stages.2.blocks.4.mlp.conv', 'vision_model.vision._en.stages.2.blocks.4.mlp.conv.bn', '_native_batch_norm_legit_no_training_8']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 814, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 814, in forward x = self.conv(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/conv_bn_act.py", line 94, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/conv_bn_act.py", line 94, in forward x = self.bn(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward x = F.batch_norm( - + getitem_24 8vision_model.vision._en.stages.2.blocks.4.mlp.fc1.weight @@ -1901,47 +1901,47 @@ getitem_24 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.4: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.4.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.2.blocks.4.mlp.fc1: torch.nn.modules.conv.Conv2d/conv2d_41: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_41 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%getitem_24, %p_vision_model_vision__en_stages_2_blocks_4_mlp_fc1_weight, %p_vision_model_vision__en_stages_2_blocks_4_mlp_fc1_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.4', 'vision_model.vision._en.stages.2.blocks.4.mlp', 'vision_model.vision._en.stages.2.blocks.4.mlp.fc1', 'conv2d_41']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.4', 'vision_model.vision._en.stages.2.blocks.4.mlp', 'vision_model.vision._en.stages.2.blocks.4.mlp.fc1', 'conv2d_41']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward x = self.fc1(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_41gelu_13 node_gelu_13"Gelu* approximate"tanhJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.4: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.4.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.2.blocks.4.mlp.act: timm.layers.activations.GELUTanh/gelu_13: aten.gelu.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'timm.layers.activations.GELUTanh', 'aten.gelu.default']J pkg.torch.onnx.fx_node%gelu_13 : [num_users=1] = call_function[target=torch.ops.aten.gelu.default](args = (%conv2d_41,), kwargs = {approximate: tanh})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.4', 'vision_model.vision._en.stages.2.blocks.4.mlp', 'vision_model.vision._en.stages.2.blocks.4.mlp.act', 'gelu_13']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.4', 'vision_model.vision._en.stages.2.blocks.4.mlp', 'vision_model.vision._en.stages.2.blocks.4.mlp.act', 'gelu_13']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, in forward x = self.act(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') - + gelu_13 8vision_model.vision._en.stages.2.blocks.4.mlp.fc2.weight 6vision_model.vision._en.stages.2.blocks.4.mlp.fc2.bias conv2d_42node_conv2d_42"Conv* @@ -1953,66 +1953,66 @@ getitem_24 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.4: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.4.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.2.blocks.4.mlp.fc2: torch.nn.modules.conv.Conv2d/conv2d_42: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_42 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%clone_16, %p_vision_model_vision__en_stages_2_blocks_4_mlp_fc2_weight, %p_vision_model_vision__en_stages_2_blocks_4_mlp_fc2_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.4', 'vision_model.vision._en.stages.2.blocks.4.mlp', 'vision_model.vision._en.stages.2.blocks.4.mlp.fc2', 'conv2d_42']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.4', 'vision_model.vision._en.stages.2.blocks.4.mlp', 'vision_model.vision._en.stages.2.blocks.4.mlp.fc2', 'conv2d_42']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward x = self.fc2(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_42 ;vision_model.vision._en.stages.2.blocks.4.layer_scale.gammamul_8 node_mul_8"MulJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.4: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.4.layer_scale: timm.models.fastvit.LayerScale2d/mul_8: aten.mul.TensorJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.LayerScale2d', 'aten.mul.Tensor']J pkg.torch.onnx.fx_node%mul_8 : [num_users=1] = call_function[target=torch.ops.aten.mul.Tensor](args = (%clone_17, %p_vision_model_vision__en_stages_2_blocks_4_layer_scale_gamma), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.4', 'vision_model.vision._en.stages.2.blocks.4.layer_scale', 'mul_8']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.4', 'vision_model.vision._en.stages.2.blocks.4.layer_scale', 'mul_8']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 646, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 646, in forward return x.mul_(self.gamma) if self.inplace else x * self.gamma - + conv2d_39 mul_8add_8 node_add_8"AddJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.4: timm.models.fastvit.RepMixerBlock/add_8: aten.add.TensorJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'aten.add.Tensor']J pkg.torch.onnx.fx_noder%add_8 : [num_users=1] = call_function[target=torch.ops.aten.add.Tensor](args = (%conv2d_39, %mul_8), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.4', 'add_8']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.4', 'add_8']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - + add_8 Ivision_model.vision._en.stages.2.blocks.5.token_mixer.reparam_conv.weight Gvision_model.vision._en.stages.2.blocks.5.token_mixer.reparam_conv.bias conv2d_43node_conv2d_43"Conv* @@ -2024,24 +2024,24 @@ Gvision_model.vision._en.stages.2.blocks.5.token_mixer.reparam_conv.bias conv2d namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.5: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.5.token_mixer: timm.models.fastvit.RepMixer/vision_model.vision._en.stages.2.blocks.5.token_mixer.reparam_conv: torch.nn.modules.conv.Conv2d/conv2d_43: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.RepMixer', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_43 : [num_users=2] = call_function[target=torch.ops.aten.conv2d.default](args = (%add_8, %p_vision_model_vision__en_stages_2_blocks_5_token_mixer_reparam_conv_weight, %p_vision_model_vision__en_stages_2_blocks_5_token_mixer_reparam_conv_bias, [1, 1], [1, 1], [1, 1], 256), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.5', 'vision_model.vision._en.stages.2.blocks.5.token_mixer', 'vision_model.vision._en.stages.2.blocks.5.token_mixer.reparam_conv', 'conv2d_43']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.5', 'vision_model.vision._en.stages.2.blocks.5.token_mixer', 'vision_model.vision._en.stages.2.blocks.5.token_mixer.reparam_conv', 'conv2d_43']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1001, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1001, in forward x = self.token_mixer(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward x = self.reparam_conv(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_43 >vision_model.vision._en.stages.2.blocks.5.mlp.conv.conv.weight Cvision_model.vision._en.stages.2.blocks.5.mlp.conv.conv.weight_bias @@ -2055,26 +2055,26 @@ getitem_27 node_Conv_200"Conv* !pkg.onnxscript.rewriter.rule_name1FuseBatchNormIntoConv, RemoveOptionalBiasFromConvJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'timm.layers.conv_bn_act.ConvNormAct', 'timm.layers.norm_act.BatchNormAct2d', 'aten._native_batch_norm_legit_no_training.default']J pkg.torch.onnx.fx_node%_native_batch_norm_legit_no_training_9 : [num_users=1] = call_function[target=torch.ops.aten._native_batch_norm_legit_no_training.default](args = (%conv2d_44, %p_vision_model_vision__en_stages_2_blocks_5_mlp_conv_bn_weight, %p_vision_model_vision__en_stages_2_blocks_5_mlp_conv_bn_bias, %b_vision_model_vision__en_stages_2_blocks_5_mlp_conv_bn_running_mean, %b_vision_model_vision__en_stages_2_blocks_5_mlp_conv_bn_running_var, 0.1, 1e-05), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.5', 'vision_model.vision._en.stages.2.blocks.5.mlp', 'vision_model.vision._en.stages.2.blocks.5.mlp.conv', 'vision_model.vision._en.stages.2.blocks.5.mlp.conv.bn', '_native_batch_norm_legit_no_training_9']J -pkg.torch.onnx.stack_trace File "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.5', 'vision_model.vision._en.stages.2.blocks.5.mlp', 'vision_model.vision._en.stages.2.blocks.5.mlp.conv', 'vision_model.vision._en.stages.2.blocks.5.mlp.conv.bn', '_native_batch_norm_legit_no_training_9']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 814, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 814, in forward x = self.conv(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/conv_bn_act.py", line 94, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/conv_bn_act.py", line 94, in forward x = self.bn(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward x = F.batch_norm( - + getitem_27 8vision_model.vision._en.stages.2.blocks.5.mlp.fc1.weight @@ -2087,47 +2087,47 @@ getitem_27 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.5: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.5.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.2.blocks.5.mlp.fc1: torch.nn.modules.conv.Conv2d/conv2d_45: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_45 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%getitem_27, %p_vision_model_vision__en_stages_2_blocks_5_mlp_fc1_weight, %p_vision_model_vision__en_stages_2_blocks_5_mlp_fc1_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.5', 'vision_model.vision._en.stages.2.blocks.5.mlp', 'vision_model.vision._en.stages.2.blocks.5.mlp.fc1', 'conv2d_45']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.5', 'vision_model.vision._en.stages.2.blocks.5.mlp', 'vision_model.vision._en.stages.2.blocks.5.mlp.fc1', 'conv2d_45']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward x = self.fc1(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_45gelu_14 node_gelu_14"Gelu* approximate"tanhJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.5: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.5.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.2.blocks.5.mlp.act: timm.layers.activations.GELUTanh/gelu_14: aten.gelu.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'timm.layers.activations.GELUTanh', 'aten.gelu.default']J pkg.torch.onnx.fx_node%gelu_14 : [num_users=1] = call_function[target=torch.ops.aten.gelu.default](args = (%conv2d_45,), kwargs = {approximate: tanh})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.5', 'vision_model.vision._en.stages.2.blocks.5.mlp', 'vision_model.vision._en.stages.2.blocks.5.mlp.act', 'gelu_14']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.5', 'vision_model.vision._en.stages.2.blocks.5.mlp', 'vision_model.vision._en.stages.2.blocks.5.mlp.act', 'gelu_14']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, in forward x = self.act(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') - + gelu_14 8vision_model.vision._en.stages.2.blocks.5.mlp.fc2.weight 6vision_model.vision._en.stages.2.blocks.5.mlp.fc2.bias conv2d_46node_conv2d_46"Conv* @@ -2139,66 +2139,66 @@ getitem_27 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.5: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.5.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.2.blocks.5.mlp.fc2: torch.nn.modules.conv.Conv2d/conv2d_46: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_46 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%clone_18, %p_vision_model_vision__en_stages_2_blocks_5_mlp_fc2_weight, %p_vision_model_vision__en_stages_2_blocks_5_mlp_fc2_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.5', 'vision_model.vision._en.stages.2.blocks.5.mlp', 'vision_model.vision._en.stages.2.blocks.5.mlp.fc2', 'conv2d_46']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.5', 'vision_model.vision._en.stages.2.blocks.5.mlp', 'vision_model.vision._en.stages.2.blocks.5.mlp.fc2', 'conv2d_46']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward x = self.fc2(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_46 ;vision_model.vision._en.stages.2.blocks.5.layer_scale.gammamul_9 node_mul_9"MulJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.5: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.2.blocks.5.layer_scale: timm.models.fastvit.LayerScale2d/mul_9: aten.mul.TensorJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.LayerScale2d', 'aten.mul.Tensor']J pkg.torch.onnx.fx_node%mul_9 : [num_users=1] = call_function[target=torch.ops.aten.mul.Tensor](args = (%clone_19, %p_vision_model_vision__en_stages_2_blocks_5_layer_scale_gamma), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.5', 'vision_model.vision._en.stages.2.blocks.5.layer_scale', 'mul_9']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.5', 'vision_model.vision._en.stages.2.blocks.5.layer_scale', 'mul_9']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 646, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 646, in forward return x.mul_(self.gamma) if self.inplace else x * self.gamma - + conv2d_43 mul_9add_9 node_add_9"AddJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.2: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.2.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.2.blocks.5: timm.models.fastvit.RepMixerBlock/add_9: aten.add.TensorJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'aten.add.Tensor']J pkg.torch.onnx.fx_noder%add_9 : [num_users=1] = call_function[target=torch.ops.aten.add.Tensor](args = (%conv2d_43, %mul_9), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.5', 'add_9']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.2', 'vision_model.vision._en.stages.2.blocks', 'vision_model.vision._en.stages.2.blocks.5', 'add_9']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - + add_9 Fvision_model.vision._en.stages.3.downsample.proj.0.reparam_conv.weight Dvision_model.vision._en.stages.3.downsample.proj.0.reparam_conv.bias conv2d_47node_conv2d_47"Conv* @@ -2210,24 +2210,24 @@ Dvision_model.vision._en.stages.3.downsample.proj.0.reparam_conv.bias conv2d_47 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.3: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.3.downsample: timm.models.fastvit.PatchEmbed/vision_model.vision._en.stages.3.downsample.proj: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.3.downsample.proj.0: timm.models.fastvit.ReparamLargeKernelConv/vision_model.vision._en.stages.3.downsample.proj.0.reparam_conv: torch.nn.modules.conv.Conv2d/conv2d_47: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'timm.models.fastvit.PatchEmbed', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.ReparamLargeKernelConv', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_47 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%add_9, %p_vision_model_vision__en_stages_3_downsample_proj_0_reparam_conv_weight, %p_vision_model_vision__en_stages_3_downsample_proj_0_reparam_conv_bias, [2, 2], [3, 3], [1, 1], 256), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.downsample', 'vision_model.vision._en.stages.3.downsample.proj', 'vision_model.vision._en.stages.3.downsample.proj.0', 'vision_model.vision._en.stages.3.downsample.proj.0.reparam_conv', 'conv2d_47']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.downsample', 'vision_model.vision._en.stages.3.downsample.proj', 'vision_model.vision._en.stages.3.downsample.proj.0', 'vision_model.vision._en.stages.3.downsample.proj.0.reparam_conv', 'conv2d_47']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1162, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1162, in forward x = self.downsample(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 628, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 628, in forward x = self.proj(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 376, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 376, in forward out = self.reparam_conv(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_47 Fvision_model.vision._en.stages.3.downsample.proj.1.reparam_conv.weight Dvision_model.vision._en.stages.3.downsample.proj.1.reparam_conv.bias conv2d_48node_conv2d_48"Conv* @@ -2239,47 +2239,47 @@ Dvision_model.vision._en.stages.3.downsample.proj.1.reparam_conv.bias conv2d_48 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.3: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.3.downsample: timm.models.fastvit.PatchEmbed/vision_model.vision._en.stages.3.downsample.proj: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.3.downsample.proj.1: timm.models.fastvit.MobileOneBlock/vision_model.vision._en.stages.3.downsample.proj.1.reparam_conv: torch.nn.modules.conv.Conv2d/conv2d_48: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'timm.models.fastvit.PatchEmbed', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.MobileOneBlock', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_48 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%conv2d_47, %p_vision_model_vision__en_stages_3_downsample_proj_1_reparam_conv_weight, %p_vision_model_vision__en_stages_3_downsample_proj_1_reparam_conv_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.downsample', 'vision_model.vision._en.stages.3.downsample.proj', 'vision_model.vision._en.stages.3.downsample.proj.1', 'vision_model.vision._en.stages.3.downsample.proj.1.reparam_conv', 'conv2d_48']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.downsample', 'vision_model.vision._en.stages.3.downsample.proj', 'vision_model.vision._en.stages.3.downsample.proj.1', 'vision_model.vision._en.stages.3.downsample.proj.1.reparam_conv', 'conv2d_48']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1162, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1162, in forward x = self.downsample(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 628, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 628, in forward x = self.proj(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward return self.act(self.se(self.reparam_conv(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_48gelu_15 node_gelu_15"Gelu* approximate"tanhJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.3: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.3.downsample: timm.models.fastvit.PatchEmbed/vision_model.vision._en.stages.3.downsample.proj: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.3.downsample.proj.1: timm.models.fastvit.MobileOneBlock/vision_model.vision._en.stages.3.downsample.proj.1.act: timm.layers.activations.GELUTanh/gelu_15: aten.gelu.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'timm.models.fastvit.PatchEmbed', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.MobileOneBlock', 'timm.layers.activations.GELUTanh', 'aten.gelu.default']J pkg.torch.onnx.fx_node%gelu_15 : [num_users=1] = call_function[target=torch.ops.aten.gelu.default](args = (%conv2d_48,), kwargs = {approximate: tanh})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.downsample', 'vision_model.vision._en.stages.3.downsample.proj', 'vision_model.vision._en.stages.3.downsample.proj.1', 'vision_model.vision._en.stages.3.downsample.proj.1.act', 'gelu_15']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.downsample', 'vision_model.vision._en.stages.3.downsample.proj', 'vision_model.vision._en.stages.3.downsample.proj.1', 'vision_model.vision._en.stages.3.downsample.proj.1.act', 'gelu_15']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1162, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1162, in forward x = self.downsample(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 628, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 628, in forward x = self.proj(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward return self.act(self.se(self.reparam_conv(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') - + gelu_15 Ivision_model.vision._en.stages.3.blocks.0.token_mixer.reparam_conv.weight Gvision_model.vision._en.stages.3.blocks.0.token_mixer.reparam_conv.bias conv2d_49node_conv2d_49"Conv* @@ -2291,24 +2291,24 @@ Gvision_model.vision._en.stages.3.blocks.0.token_mixer.reparam_conv.bias conv2d namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.3: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.3.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.3.blocks.0: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.3.blocks.0.token_mixer: timm.models.fastvit.RepMixer/vision_model.vision._en.stages.3.blocks.0.token_mixer.reparam_conv: torch.nn.modules.conv.Conv2d/conv2d_49: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.RepMixer', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_49 : [num_users=2] = call_function[target=torch.ops.aten.conv2d.default](args = (%gelu_15, %p_vision_model_vision__en_stages_3_blocks_0_token_mixer_reparam_conv_weight, %p_vision_model_vision__en_stages_3_blocks_0_token_mixer_reparam_conv_bias, [1, 1], [1, 1], [1, 1], 512), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.0', 'vision_model.vision._en.stages.3.blocks.0.token_mixer', 'vision_model.vision._en.stages.3.blocks.0.token_mixer.reparam_conv', 'conv2d_49']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.0', 'vision_model.vision._en.stages.3.blocks.0.token_mixer', 'vision_model.vision._en.stages.3.blocks.0.token_mixer.reparam_conv', 'conv2d_49']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1001, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1001, in forward x = self.token_mixer(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward x = self.reparam_conv(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_49 >vision_model.vision._en.stages.3.blocks.0.mlp.conv.conv.weight Cvision_model.vision._en.stages.3.blocks.0.mlp.conv.conv.weight_bias @@ -2322,26 +2322,26 @@ getitem_30 node_Conv_202"Conv* !pkg.onnxscript.rewriter.rule_name1FuseBatchNormIntoConv, RemoveOptionalBiasFromConvJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'timm.layers.conv_bn_act.ConvNormAct', 'timm.layers.norm_act.BatchNormAct2d', 'aten._native_batch_norm_legit_no_training.default']J pkg.torch.onnx.fx_node%_native_batch_norm_legit_no_training_10 : [num_users=1] = call_function[target=torch.ops.aten._native_batch_norm_legit_no_training.default](args = (%conv2d_50, %p_vision_model_vision__en_stages_3_blocks_0_mlp_conv_bn_weight, %p_vision_model_vision__en_stages_3_blocks_0_mlp_conv_bn_bias, %b_vision_model_vision__en_stages_3_blocks_0_mlp_conv_bn_running_mean, %b_vision_model_vision__en_stages_3_blocks_0_mlp_conv_bn_running_var, 0.1, 1e-05), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.0', 'vision_model.vision._en.stages.3.blocks.0.mlp', 'vision_model.vision._en.stages.3.blocks.0.mlp.conv', 'vision_model.vision._en.stages.3.blocks.0.mlp.conv.bn', '_native_batch_norm_legit_no_training_10']J -pkg.torch.onnx.stack_trace File "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.0', 'vision_model.vision._en.stages.3.blocks.0.mlp', 'vision_model.vision._en.stages.3.blocks.0.mlp.conv', 'vision_model.vision._en.stages.3.blocks.0.mlp.conv.bn', '_native_batch_norm_legit_no_training_10']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 814, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 814, in forward x = self.conv(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/conv_bn_act.py", line 94, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/conv_bn_act.py", line 94, in forward x = self.bn(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward x = F.batch_norm( - + getitem_30 8vision_model.vision._en.stages.3.blocks.0.mlp.fc1.weight @@ -2354,47 +2354,47 @@ getitem_30 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.3: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.3.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.3.blocks.0: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.3.blocks.0.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.3.blocks.0.mlp.fc1: torch.nn.modules.conv.Conv2d/conv2d_51: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_51 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%getitem_30, %p_vision_model_vision__en_stages_3_blocks_0_mlp_fc1_weight, %p_vision_model_vision__en_stages_3_blocks_0_mlp_fc1_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.0', 'vision_model.vision._en.stages.3.blocks.0.mlp', 'vision_model.vision._en.stages.3.blocks.0.mlp.fc1', 'conv2d_51']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.0', 'vision_model.vision._en.stages.3.blocks.0.mlp', 'vision_model.vision._en.stages.3.blocks.0.mlp.fc1', 'conv2d_51']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward x = self.fc1(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_51gelu_16 node_gelu_16"Gelu* approximate"tanhJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.3: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.3.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.3.blocks.0: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.3.blocks.0.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.3.blocks.0.mlp.act: timm.layers.activations.GELUTanh/gelu_16: aten.gelu.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'timm.layers.activations.GELUTanh', 'aten.gelu.default']J pkg.torch.onnx.fx_node%gelu_16 : [num_users=1] = call_function[target=torch.ops.aten.gelu.default](args = (%conv2d_51,), kwargs = {approximate: tanh})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.0', 'vision_model.vision._en.stages.3.blocks.0.mlp', 'vision_model.vision._en.stages.3.blocks.0.mlp.act', 'gelu_16']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.0', 'vision_model.vision._en.stages.3.blocks.0.mlp', 'vision_model.vision._en.stages.3.blocks.0.mlp.act', 'gelu_16']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, in forward x = self.act(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') - + gelu_16 8vision_model.vision._en.stages.3.blocks.0.mlp.fc2.weight 6vision_model.vision._en.stages.3.blocks.0.mlp.fc2.bias conv2d_52node_conv2d_52"Conv* @@ -2406,64 +2406,64 @@ getitem_30 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.3: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.3.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.3.blocks.0: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.3.blocks.0.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.3.blocks.0.mlp.fc2: torch.nn.modules.conv.Conv2d/conv2d_52: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_52 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%clone_20, %p_vision_model_vision__en_stages_3_blocks_0_mlp_fc2_weight, %p_vision_model_vision__en_stages_3_blocks_0_mlp_fc2_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.0', 'vision_model.vision._en.stages.3.blocks.0.mlp', 'vision_model.vision._en.stages.3.blocks.0.mlp.fc2', 'conv2d_52']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.0', 'vision_model.vision._en.stages.3.blocks.0.mlp', 'vision_model.vision._en.stages.3.blocks.0.mlp.fc2', 'conv2d_52']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward x = self.fc2(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_52 ;vision_model.vision._en.stages.3.blocks.0.layer_scale.gammamul_10 node_mul_10"MulJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.3: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.3.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.3.blocks.0: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.3.blocks.0.layer_scale: timm.models.fastvit.LayerScale2d/mul_10: aten.mul.TensorJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.LayerScale2d', 'aten.mul.Tensor']J pkg.torch.onnx.fx_node%mul_10 : [num_users=1] = call_function[target=torch.ops.aten.mul.Tensor](args = (%clone_21, %p_vision_model_vision__en_stages_3_blocks_0_layer_scale_gamma), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.0', 'vision_model.vision._en.stages.3.blocks.0.layer_scale', 'mul_10']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.0', 'vision_model.vision._en.stages.3.blocks.0.layer_scale', 'mul_10']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 646, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 646, in forward return x.mul_(self.gamma) if self.inplace else x * self.gamma - + conv2d_49 mul_10add_10 node_add_10"AddJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.3: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.3.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.3.blocks.0: timm.models.fastvit.RepMixerBlock/add_10: aten.add.TensorJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'aten.add.Tensor']J pkg.torch.onnx.fx_nodet%add_10 : [num_users=1] = call_function[target=torch.ops.aten.add.Tensor](args = (%conv2d_49, %mul_10), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.0', 'add_10']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.0', 'add_10']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - + add_10 Ivision_model.vision._en.stages.3.blocks.1.token_mixer.reparam_conv.weight Gvision_model.vision._en.stages.3.blocks.1.token_mixer.reparam_conv.bias conv2d_53node_conv2d_53"Conv* @@ -2475,24 +2475,24 @@ Gvision_model.vision._en.stages.3.blocks.1.token_mixer.reparam_conv.bias conv2d namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.3: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.3.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.3.blocks.1: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.3.blocks.1.token_mixer: timm.models.fastvit.RepMixer/vision_model.vision._en.stages.3.blocks.1.token_mixer.reparam_conv: torch.nn.modules.conv.Conv2d/conv2d_53: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.RepMixer', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_53 : [num_users=2] = call_function[target=torch.ops.aten.conv2d.default](args = (%add_10, %p_vision_model_vision__en_stages_3_blocks_1_token_mixer_reparam_conv_weight, %p_vision_model_vision__en_stages_3_blocks_1_token_mixer_reparam_conv_bias, [1, 1], [1, 1], [1, 1], 512), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.1', 'vision_model.vision._en.stages.3.blocks.1.token_mixer', 'vision_model.vision._en.stages.3.blocks.1.token_mixer.reparam_conv', 'conv2d_53']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.1', 'vision_model.vision._en.stages.3.blocks.1.token_mixer', 'vision_model.vision._en.stages.3.blocks.1.token_mixer.reparam_conv', 'conv2d_53']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1001, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1001, in forward x = self.token_mixer(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward x = self.reparam_conv(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_53 >vision_model.vision._en.stages.3.blocks.1.mlp.conv.conv.weight Cvision_model.vision._en.stages.3.blocks.1.mlp.conv.conv.weight_bias @@ -2506,26 +2506,26 @@ getitem_33 node_Conv_204"Conv* !pkg.onnxscript.rewriter.rule_name1FuseBatchNormIntoConv, RemoveOptionalBiasFromConvJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'timm.layers.conv_bn_act.ConvNormAct', 'timm.layers.norm_act.BatchNormAct2d', 'aten._native_batch_norm_legit_no_training.default']J pkg.torch.onnx.fx_node%_native_batch_norm_legit_no_training_11 : [num_users=1] = call_function[target=torch.ops.aten._native_batch_norm_legit_no_training.default](args = (%conv2d_54, %p_vision_model_vision__en_stages_3_blocks_1_mlp_conv_bn_weight, %p_vision_model_vision__en_stages_3_blocks_1_mlp_conv_bn_bias, %b_vision_model_vision__en_stages_3_blocks_1_mlp_conv_bn_running_mean, %b_vision_model_vision__en_stages_3_blocks_1_mlp_conv_bn_running_var, 0.1, 1e-05), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.1', 'vision_model.vision._en.stages.3.blocks.1.mlp', 'vision_model.vision._en.stages.3.blocks.1.mlp.conv', 'vision_model.vision._en.stages.3.blocks.1.mlp.conv.bn', '_native_batch_norm_legit_no_training_11']J -pkg.torch.onnx.stack_trace File "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.1', 'vision_model.vision._en.stages.3.blocks.1.mlp', 'vision_model.vision._en.stages.3.blocks.1.mlp.conv', 'vision_model.vision._en.stages.3.blocks.1.mlp.conv.bn', '_native_batch_norm_legit_no_training_11']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 814, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 814, in forward x = self.conv(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/conv_bn_act.py", line 94, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/conv_bn_act.py", line 94, in forward x = self.bn(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward x = F.batch_norm( - + getitem_33 8vision_model.vision._en.stages.3.blocks.1.mlp.fc1.weight @@ -2538,47 +2538,47 @@ getitem_33 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.3: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.3.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.3.blocks.1: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.3.blocks.1.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.3.blocks.1.mlp.fc1: torch.nn.modules.conv.Conv2d/conv2d_55: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_55 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%getitem_33, %p_vision_model_vision__en_stages_3_blocks_1_mlp_fc1_weight, %p_vision_model_vision__en_stages_3_blocks_1_mlp_fc1_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.1', 'vision_model.vision._en.stages.3.blocks.1.mlp', 'vision_model.vision._en.stages.3.blocks.1.mlp.fc1', 'conv2d_55']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.1', 'vision_model.vision._en.stages.3.blocks.1.mlp', 'vision_model.vision._en.stages.3.blocks.1.mlp.fc1', 'conv2d_55']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward x = self.fc1(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_55gelu_17 node_gelu_17"Gelu* approximate"tanhJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.3: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.3.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.3.blocks.1: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.3.blocks.1.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.3.blocks.1.mlp.act: timm.layers.activations.GELUTanh/gelu_17: aten.gelu.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'timm.layers.activations.GELUTanh', 'aten.gelu.default']J pkg.torch.onnx.fx_node%gelu_17 : [num_users=1] = call_function[target=torch.ops.aten.gelu.default](args = (%conv2d_55,), kwargs = {approximate: tanh})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.1', 'vision_model.vision._en.stages.3.blocks.1.mlp', 'vision_model.vision._en.stages.3.blocks.1.mlp.act', 'gelu_17']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.1', 'vision_model.vision._en.stages.3.blocks.1.mlp', 'vision_model.vision._en.stages.3.blocks.1.mlp.act', 'gelu_17']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, in forward x = self.act(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') - + gelu_17 8vision_model.vision._en.stages.3.blocks.1.mlp.fc2.weight 6vision_model.vision._en.stages.3.blocks.1.mlp.fc2.bias conv2d_56node_conv2d_56"Conv* @@ -2590,64 +2590,64 @@ getitem_33 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.3: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.3.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.3.blocks.1: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.3.blocks.1.mlp: timm.models.fastvit.ConvMlp/vision_model.vision._en.stages.3.blocks.1.mlp.fc2: torch.nn.modules.conv.Conv2d/conv2d_56: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.ConvMlp', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_56 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%clone_22, %p_vision_model_vision__en_stages_3_blocks_1_mlp_fc2_weight, %p_vision_model_vision__en_stages_3_blocks_1_mlp_fc2_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.1', 'vision_model.vision._en.stages.3.blocks.1.mlp', 'vision_model.vision._en.stages.3.blocks.1.mlp.fc2', 'conv2d_56']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.1', 'vision_model.vision._en.stages.3.blocks.1.mlp', 'vision_model.vision._en.stages.3.blocks.1.mlp.fc2', 'conv2d_56']J +pkg.torch.onnx.stack_trace File "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward x = self.fc2(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_56 ;vision_model.vision._en.stages.3.blocks.1.layer_scale.gammamul_11 node_mul_11"MulJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.3: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.3.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.3.blocks.1: timm.models.fastvit.RepMixerBlock/vision_model.vision._en.stages.3.blocks.1.layer_scale: timm.models.fastvit.LayerScale2d/mul_11: aten.mul.TensorJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'timm.models.fastvit.LayerScale2d', 'aten.mul.Tensor']J pkg.torch.onnx.fx_node%mul_11 : [num_users=1] = call_function[target=torch.ops.aten.mul.Tensor](args = (%clone_23, %p_vision_model_vision__en_stages_3_blocks_1_layer_scale_gamma), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.1', 'vision_model.vision._en.stages.3.blocks.1.layer_scale', 'mul_11']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.1', 'vision_model.vision._en.stages.3.blocks.1.layer_scale', 'mul_11']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 646, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 646, in forward return x.mul_(self.gamma) if self.inplace else x * self.gamma - + conv2d_53 mul_11add_11 node_add_11"AddJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.stages.3: timm.models.fastvit.FastVitStage/vision_model.vision._en.stages.3.blocks: torch.nn.modules.container.Sequential/vision_model.vision._en.stages.3.blocks.1: timm.models.fastvit.RepMixerBlock/add_11: aten.add.TensorJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.FastVitStage', 'torch.nn.modules.container.Sequential', 'timm.models.fastvit.RepMixerBlock', 'aten.add.Tensor']J pkg.torch.onnx.fx_nodet%add_11 : [num_users=1] = call_function[target=torch.ops.aten.add.Tensor](args = (%conv2d_53, %mul_11), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.1', 'add_11']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.stages.3', 'vision_model.vision._en.stages.3.blocks', 'vision_model.vision._en.stages.3.blocks.1', 'add_11']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1167, in forward x = self.blocks(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x))) - + add_11 6vision_model.vision._en.final_conv.reparam_conv.weight 4vision_model.vision._en.final_conv.reparam_conv.bias conv2d_57node_conv2d_57"Conv* @@ -2659,18 +2659,18 @@ getitem_33 namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.final_conv: timm.models.fastvit.MobileOneBlock/vision_model.vision._en.final_conv.reparam_conv: torch.nn.modules.conv.Conv2d/conv2d_57: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.MobileOneBlock', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_57 : [num_users=2] = call_function[target=torch.ops.aten.conv2d.default](args = (%add_11, %p_vision_model_vision__en_final_conv_reparam_conv_weight, %p_vision_model_vision__en_final_conv_reparam_conv_bias, [1, 1], [1, 1], [1, 1], 512), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.final_conv', 'vision_model.vision._en.final_conv.reparam_conv', 'conv2d_57']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.final_conv', 'vision_model.vision._en.final_conv.reparam_conv', 'conv2d_57']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward return self.act(self.se(self.reparam_conv(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_57 val_137mean node_mean" ReduceMean* @@ -2679,18 +2679,18 @@ ReduceMean* namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.final_conv: timm.models.fastvit.MobileOneBlock/vision_model.vision._en.final_conv.se: timm.layers.squeeze_excite.SEModule/mean: aten.mean.dimJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.MobileOneBlock', 'timm.layers.squeeze_excite.SEModule', 'aten.mean.dim']J pkg.torch.onnx.fx_nodeu%mean : [num_users=1] = call_function[target=torch.ops.aten.mean.dim](args = (%conv2d_57, [2, 3], True), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.final_conv', 'vision_model.vision._en.final_conv.se', 'mean']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.final_conv', 'vision_model.vision._en.final_conv.se', 'mean']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward return self.act(self.se(self.reparam_conv(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/squeeze_excite.py", line 56, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/squeeze_excite.py", line 56, in forward x_se = x.mean((2, 3), keepdim=True) - + mean 0vision_model.vision._en.final_conv.se.fc1.weight .vision_model.vision._en.final_conv.se.fc1.bias conv2d_58node_conv2d_58"Conv* @@ -2702,38 +2702,38 @@ ReduceMean* namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.final_conv: timm.models.fastvit.MobileOneBlock/vision_model.vision._en.final_conv.se: timm.layers.squeeze_excite.SEModule/vision_model.vision._en.final_conv.se.fc1: torch.nn.modules.conv.Conv2d/conv2d_58: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.MobileOneBlock', 'timm.layers.squeeze_excite.SEModule', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_58 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%mean, %p_vision_model_vision__en_final_conv_se_fc1_weight, %p_vision_model_vision__en_final_conv_se_fc1_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.final_conv', 'vision_model.vision._en.final_conv.se', 'vision_model.vision._en.final_conv.se.fc1', 'conv2d_58']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.final_conv', 'vision_model.vision._en.final_conv.se', 'vision_model.vision._en.final_conv.se.fc1', 'conv2d_58']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward return self.act(self.se(self.reparam_conv(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/squeeze_excite.py", line 60, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/squeeze_excite.py", line 60, in forward x_se = self.fc1(x_se) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_58relu node_relu"ReluJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.final_conv: timm.models.fastvit.MobileOneBlock/vision_model.vision._en.final_conv.se: timm.layers.squeeze_excite.SEModule/vision_model.vision._en.final_conv.se.act: torch.nn.modules.activation.ReLU/relu: aten.relu.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.MobileOneBlock', 'timm.layers.squeeze_excite.SEModule', 'torch.nn.modules.activation.ReLU', 'aten.relu.default']J pkg.torch.onnx.fx_nodel%relu : [num_users=1] = call_function[target=torch.ops.aten.relu.default](args = (%conv2d_58,), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.final_conv', 'vision_model.vision._en.final_conv.se', 'vision_model.vision._en.final_conv.se.act', 'relu']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.final_conv', 'vision_model.vision._en.final_conv.se', 'vision_model.vision._en.final_conv.se.act', 'relu']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward return self.act(self.se(self.reparam_conv(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/squeeze_excite.py", line 61, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/squeeze_excite.py", line 61, in forward x_se = self.act(self.bn(x_se)) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/activation.py", line 143, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/activation.py", line 143, in forward return F.relu(input, inplace=self.inplace) - + relu 0vision_model.vision._en.final_conv.se.fc2.weight .vision_model.vision._en.final_conv.se.fc2.bias conv2d_59node_conv2d_59"Conv* @@ -2745,72 +2745,72 @@ ReduceMean* namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.final_conv: timm.models.fastvit.MobileOneBlock/vision_model.vision._en.final_conv.se: timm.layers.squeeze_excite.SEModule/vision_model.vision._en.final_conv.se.fc2: torch.nn.modules.conv.Conv2d/conv2d_59: aten.conv2d.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.MobileOneBlock', 'timm.layers.squeeze_excite.SEModule', 'torch.nn.modules.conv.Conv2d', 'aten.conv2d.default']J pkg.torch.onnx.fx_node%conv2d_59 : [num_users=1] = call_function[target=torch.ops.aten.conv2d.default](args = (%relu, %p_vision_model_vision__en_final_conv_se_fc2_weight, %p_vision_model_vision__en_final_conv_se_fc2_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.final_conv', 'vision_model.vision._en.final_conv.se', 'vision_model.vision._en.final_conv.se.fc2', 'conv2d_59']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.final_conv', 'vision_model.vision._en.final_conv.se', 'vision_model.vision._en.final_conv.se.fc2', 'conv2d_59']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward return self.act(self.se(self.reparam_conv(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/squeeze_excite.py", line 62, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/squeeze_excite.py", line 62, in forward x_se = self.fc2(x_se) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward return self._conv_forward(input, self.weight, self.bias) - + conv2d_59sigmoid node_sigmoid"SigmoidJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.final_conv: timm.models.fastvit.MobileOneBlock/vision_model.vision._en.final_conv.se: timm.layers.squeeze_excite.SEModule/vision_model.vision._en.final_conv.se.gate: timm.layers.activations.Sigmoid/sigmoid: aten.sigmoid.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.MobileOneBlock', 'timm.layers.squeeze_excite.SEModule', 'timm.layers.activations.Sigmoid', 'aten.sigmoid.default']J pkg.torch.onnx.fx_noder%sigmoid : [num_users=1] = call_function[target=torch.ops.aten.sigmoid.default](args = (%conv2d_59,), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.final_conv', 'vision_model.vision._en.final_conv.se', 'vision_model.vision._en.final_conv.se.gate', 'sigmoid']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.final_conv', 'vision_model.vision._en.final_conv.se', 'vision_model.vision._en.final_conv.se.gate', 'sigmoid']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward return self.act(self.se(self.reparam_conv(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/squeeze_excite.py", line 63, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/squeeze_excite.py", line 63, in forward return x * self.gate(x_se) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 57, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 57, in forward return x.sigmoid_() if self.inplace else x.sigmoid() - + conv2d_57 sigmoidmul_12 node_mul_12"MulJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.final_conv: timm.models.fastvit.MobileOneBlock/vision_model.vision._en.final_conv.se: timm.layers.squeeze_excite.SEModule/mul_12: aten.mul.TensorJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.MobileOneBlock', 'timm.layers.squeeze_excite.SEModule', 'aten.mul.Tensor']J pkg.torch.onnx.fx_nodeu%mul_12 : [num_users=1] = call_function[target=torch.ops.aten.mul.Tensor](args = (%conv2d_57, %sigmoid), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.final_conv', 'vision_model.vision._en.final_conv.se', 'mul_12']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.final_conv', 'vision_model.vision._en.final_conv.se', 'mul_12']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward return self.act(self.se(self.reparam_conv(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/squeeze_excite.py", line 63, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/squeeze_excite.py", line 63, in forward return x * self.gate(x_se) - + mul_12gelu_18 node_gelu_18"Gelu* approximate"tanhJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.final_conv: timm.models.fastvit.MobileOneBlock/vision_model.vision._en.final_conv.act: timm.layers.activations.GELUTanh/gelu_18: aten.gelu.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.models.fastvit.MobileOneBlock', 'timm.layers.activations.GELUTanh', 'aten.gelu.default']J pkg.torch.onnx.fx_node}%gelu_18 : [num_users=1] = call_function[target=torch.ops.aten.gelu.default](args = (%mul_12,), kwargs = {approximate: tanh})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.final_conv', 'vision_model.vision._en.final_conv.act', 'gelu_18']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.final_conv', 'vision_model.vision._en.final_conv.act', 'gelu_18']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward x = self.forward_features(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward return self.act(self.se(self.reparam_conv(x))) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') - + gelu_18 val_140mean_1 node_mean_1" ReduceMean* @@ -2819,40 +2819,40 @@ ReduceMean* namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.head: timm.layers.classifier.ClassifierHead/vision_model.vision._en.head.global_pool: timm.layers.adaptive_avgmax_pool.SelectAdaptivePool2d/vision_model.vision._en.head.global_pool.pool: torch.nn.modules.pooling.AdaptiveAvgPool2d/mean_1: aten.mean.dimJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.layers.classifier.ClassifierHead', 'timm.layers.adaptive_avgmax_pool.SelectAdaptivePool2d', 'torch.nn.modules.pooling.AdaptiveAvgPool2d', 'aten.mean.dim']J pkg.torch.onnx.fx_nodew%mean_1 : [num_users=1] = call_function[target=torch.ops.aten.mean.dim](args = (%gelu_18, [-1, -2], True), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.head', 'vision_model.vision._en.head.global_pool', 'vision_model.vision._en.head.global_pool.pool', 'mean_1']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.head', 'vision_model.vision._en.head.global_pool', 'vision_model.vision._en.head.global_pool.pool', 'mean_1']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1421, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1421, in forward x = self.forward_head(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/classifier.py", line 137, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/classifier.py", line 137, in forward x = self.global_pool(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/adaptive_avgmax_pool.py", line 172, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/adaptive_avgmax_pool.py", line 172, in forward x = self.pool(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/pooling.py", line 1510, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/pooling.py", line 1510, in forward return F.adaptive_avg_pool2d(input, self.output_size) - + mean_1 val_144view node_view"Reshape* allowzeroJ namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.head: timm.layers.classifier.ClassifierHead/vision_model.vision._en.head.global_pool: timm.layers.adaptive_avgmax_pool.SelectAdaptivePool2d/vision_model.vision._en.head.global_pool.flatten: torch.nn.modules.flatten.Flatten/view: aten.view.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.layers.classifier.ClassifierHead', 'timm.layers.adaptive_avgmax_pool.SelectAdaptivePool2d', 'torch.nn.modules.flatten.Flatten', 'aten.view.default']J pkg.torch.onnx.fx_nodes%view : [num_users=1] = call_function[target=torch.ops.aten.view.default](args = (%mean_1, [1, 1024]), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.head', 'vision_model.vision._en.head.global_pool', 'vision_model.vision._en.head.global_pool.flatten', 'view']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.head', 'vision_model.vision._en.head.global_pool', 'vision_model.vision._en.head.global_pool.flatten', 'view']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1421, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1421, in forward x = self.forward_head(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/classifier.py", line 137, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/classifier.py", line 137, in forward x = self.global_pool(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/adaptive_avgmax_pool.py", line 173, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/adaptive_avgmax_pool.py", line 173, in forward x = self.flatten(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/flatten.py", line 55, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/flatten.py", line 55, in forward return input.flatten(self.start_dim, self.end_dim) - + view &vision_model.vision._en.head.fc.weight $vision_model.vision._en.head.fc.biaslinear node_linear"Gemm* @@ -2863,18 +2863,18 @@ $vision_model.vision._en.head.fc.biaslinear node_linear"Gemm* namespace: __main__.FlattenedVisionModel/vision_model.vision: xx.training.path.supercombo.Vision/vision_model.vision._en: timm.models.fastvit.FastVit/vision_model.vision._en.head: timm.layers.classifier.ClassifierHead/vision_model.vision._en.head.fc: torch.nn.modules.linear.Linear/linear: aten.linear.defaultJ pkg.torch.onnx.class_hierarchy['__main__.FlattenedVisionModel', 'xx.training.path.supercombo.Vision', 'timm.models.fastvit.FastVit', 'timm.layers.classifier.ClassifierHead', 'torch.nn.modules.linear.Linear', 'aten.linear.default']J pkg.torch.onnx.fx_node%linear : [num_users=3] = call_function[target=torch.ops.aten.linear.default](args = (%clone_24, %p_vision_model_vision__en_head_fc_weight, %p_vision_model_vision__en_head_fc_bias), kwargs = {})J -pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.head', 'vision_model.vision._en.head.fc', 'linear']J -pkg.torch.onnx.stack_traceFile "/home/batman/xx/./ml_tools/openpilot_compile/compile_supercombo.py", line 91, in forward +pkg.torch.onnx.name_scopes['', 'vision_model.vision', 'vision_model.vision._en', 'vision_model.vision._en.head', 'vision_model.vision._en.head.fc', 'linear']J +pkg.torch.onnx.stack_traceFile "/home/batman/xx3/ml_tools/openpilot_compile/compile_supercombo.py", line 82, in forward return torch.cat([*self.forward_dict(inputs).values(), self.pad], dim=1) - File "/home/batman/xx/xx/training/path/supercombo.py", line 185, in forward + File "/home/batman/xx3/xx/training/path/supercombo.py", line 185, in forward return self._en(x).flatten(1) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1421, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1421, in forward x = self.forward_head(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/classifier.py", line 141, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/classifier.py", line 141, in forward x = self.fc(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/torch/nn/modules/linear.py", line 134, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/linear.py", line 134, in forward return F.linear(input, self.weight, self.bias) - + linear >vision_model.point_policy.summarizer.resblock.block_a.0.weight vision_model.point_policy.summarizer.resblock.block_a.1.weight vision_model.point_policy.summarizer.resblock.block_a.3.weight vision_model.point_policy.summarizer.resblock.block_a.4.weight vision_model.point_policy.summarizer.resblock.block_b.1.weight vision_model.point_policy.summarizer.resblock.block_b.2.weight vision_model.point_policy.summarizer.resblock.block_b.4.weight vision_model.point_policy.summarizer.resblock.block_b.5.weight vision_model.point_policy.hydra.in_layer.road_transform.weight vision_model.point_policy.hydra.res_layer.desire_pred.0.weight vision_model.point_policy.hydra.res_layer.desire_pred.2.weight vision_model.point_policy.hydra.final_layer.desire_pred.weight