diff --git a/uncompiledmodels/driving_off_policy.onnx b/uncompiledmodels/driving_off_policy.onnx index ad1290daf..0d0863500 100644 Binary files a/uncompiledmodels/driving_off_policy.onnx and b/uncompiledmodels/driving_off_policy.onnx differ diff --git a/uncompiledmodels/driving_on_policy.onnx b/uncompiledmodels/driving_on_policy.onnx index 4af859440..2b9c200ed 100644 Binary files a/uncompiledmodels/driving_on_policy.onnx and b/uncompiledmodels/driving_on_policy.onnx differ diff --git a/uncompiledmodels/driving_vision.onnx b/uncompiledmodels/driving_vision.onnx index cd72167e6..73a3ab92d 100644 --- a/uncompiledmodels/driving_vision.onnx +++ b/uncompiledmodels/driving_vision.onnx @@ -1,5 +1,5 @@  -pytorch 2.10.0+cu130:Lj +pytorch 2.10.0+cu130:  img_to_copy node__to_copy"Cast* to @@ -32,7 +32,7 @@ _to_copy_1catnode_cat"Concat* pkg.torch.onnx.name_scopes"['', 'vision_model.vision', 'cat']J pkg.torch.onnx.stack_traceFile "/home/batman/xx/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 190, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 195, in forward x = torch.cat([x[name] for name in self.config.input_frame_names], dim=1)  cat @@ -43,7 +43,7 @@ _to_copy_1catnode_cat"Concat* 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)}  sub @@ -54,7 +54,7 @@ _to_copy_1catnode_cat"Concat* 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)}  div @@ -71,9 +71,9 @@ _to_copy_1catnode_cat"Concat* 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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 input = module(input) @@ -90,9 +90,9 @@ _to_copy_1catnode_cat"Concat* 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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 input = module(input) @@ -115,9 +115,9 @@ _to_copy_1catnode_cat"Concat* 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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 input = module(input) @@ -134,9 +134,9 @@ _to_copy_1catnode_cat"Concat* 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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 input = module(input) @@ -159,9 +159,9 @@ _to_copy_1catnode_cat"Concat* 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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 input = module(input) @@ -178,9 +178,9 @@ _to_copy_1catnode_cat"Concat* 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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 input = module(input) @@ -203,53 +203,53 @@ Gvision_model.vision._en.stages.0.blocks.0.token_mixer.reparam_conv.biasconv2d 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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 - x = self.blocks(x) - File "/home/batman/xx/.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 - x = self.token_mixer(x) - File "/home/batman/xx/.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 - 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* -group@* -pads@@@@* -auto_pad"NOTSET* -strides@@* - dilations@@J - 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.conv: timm.layers.conv_bn_act.ConvNormAct/vision_model.vision._en.stages.0.blocks.0.mlp.conv.bn: timm.layers.norm_act.BatchNormAct2d/_native_batch_norm_legit_no_training: aten._native_batch_norm_legit_no_training.defaultJV -!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.01, 0.001), 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 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 191, in forward - return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + x = self.token_mixer(x) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 718, 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 + 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_283"Conv* +group@* +pads@@@@* +auto_pad"NOTSET* +strides@@* + dilations@@J + 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.conv: timm.layers.conv_bn_act.ConvNormAct/vision_model.vision._en.stages.0.blocks.0.mlp.conv.bn: xx.training.path.allnorm.AllNormAct2d/_native_batch_norm_legit_no_training: aten._native_batch_norm_legit_no_training.defaultJV +!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', 'xx.training.path.allnorm.AllNormAct2d', '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, %repeat_2, %repeat_3, %repeat, %repeat_1, 0.01, 0.001), 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 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 196, in forward + return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, in forward + x = self.forward_features(x) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 + input = module(input) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, 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 x = self.bn(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward - x = F.batch_norm( + File "/home/batman/xx/xx/training/path/allnorm.py", line 60, in forward + return self.act(self.drop(self._allnorm_forward(x)))  getitem 8vision_model.vision._en.stages.0.blocks.0.mlp.fc1.weight @@ -265,17 +265,17 @@ Cvision_model.vision._en.stages.0.blocks.0.mlp.conv.conv.weight_biasgetitem n 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/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, 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 return self._conv_forward(input, self.weight, self.bias) @@ -288,17 +288,17 @@ Cvision_model.vision._en.stages.0.blocks.0.mlp.conv.conv.weight_biasgetitem n 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 817, in forward x = self.act(x) File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') @@ -317,17 +317,17 @@ Cvision_model.vision._en.stages.0.blocks.0.mlp.conv.conv.weight_biasgetitem n 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/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 819, 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 return self._conv_forward(input, self.weight, self.bias) @@ -340,17 +340,17 @@ Cvision_model.vision._en.stages.0.blocks.0.mlp.conv.conv.weight_biasgetitem n 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 647, in forward return x.mul_(self.gamma) if self.inplace else x * self.gamma  conv2d_3 @@ -361,15 +361,15 @@ Cvision_model.vision._en.stages.0.blocks.0.mlp.conv.conv.weight_biasgetitem n 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x)))  add @@ -386,53 +386,53 @@ Gvision_model.vision._en.stages.0.blocks.1.token_mixer.reparam_conv.biasconv2d 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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 - x = self.blocks(x) - File "/home/batman/xx/.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 - x = self.token_mixer(x) - File "/home/batman/xx/.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 - 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* -group@* -pads@@@@* -auto_pad"NOTSET* -strides@@* - dilations@@J - 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.conv: timm.layers.conv_bn_act.ConvNormAct/vision_model.vision._en.stages.0.blocks.1.mlp.conv.bn: timm.layers.norm_act.BatchNormAct2d/_native_batch_norm_legit_no_training_1: aten._native_batch_norm_legit_no_training.defaultJV -!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.01, 0.001), 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 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 191, in forward - return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + x = self.token_mixer(x) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 718, 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 + 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_285"Conv* +group@* +pads@@@@* +auto_pad"NOTSET* +strides@@* + dilations@@J + 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.conv: timm.layers.conv_bn_act.ConvNormAct/vision_model.vision._en.stages.0.blocks.1.mlp.conv.bn: xx.training.path.allnorm.AllNormAct2d/_native_batch_norm_legit_no_training_1: aten._native_batch_norm_legit_no_training.defaultJV +!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', 'xx.training.path.allnorm.AllNormAct2d', '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, %repeat_6, %repeat_7, %repeat_4, %repeat_5, 0.01, 0.001), 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 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 196, in forward + return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, in forward + x = self.forward_features(x) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 + input = module(input) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, 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 x = self.bn(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward - x = F.batch_norm( + File "/home/batman/xx/xx/training/path/allnorm.py", line 60, in forward + return self.act(self.drop(self._allnorm_forward(x)))  getitem_3 8vision_model.vision._en.stages.0.blocks.1.mlp.fc1.weight @@ -448,17 +448,17 @@ Cvision_model.vision._en.stages.0.blocks.1.mlp.conv.conv.weight_bias getitem_3 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/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, 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 return self._conv_forward(input, self.weight, self.bias) @@ -471,17 +471,17 @@ Cvision_model.vision._en.stages.0.blocks.1.mlp.conv.conv.weight_bias getitem_3 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 817, in forward x = self.act(x) File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') @@ -500,17 +500,17 @@ Cvision_model.vision._en.stages.0.blocks.1.mlp.conv.conv.weight_bias getitem_3 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/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 819, 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 return self._conv_forward(input, self.weight, self.bias) @@ -524,17 +524,17 @@ node_mul_1"MulJ 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 647, in forward return x.mul_(self.gamma) if self.inplace else x * self.gamma  conv2d_7 @@ -546,15 +546,15 @@ node_add_1"AddJ 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x)))  add_1 @@ -571,17 +571,17 @@ Dvision_model.vision._en.stages.1.downsample.proj.0.reparam_conv.bias conv2d_11 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1163, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 629, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 376, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 377, 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 return self._conv_forward(input, self.weight, self.bias) @@ -600,13 +600,13 @@ Dvision_model.vision._en.stages.1.downsample.proj.1.reparam_conv.bias conv2d_12 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1163, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 629, 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 input = module(input) @@ -623,13 +623,13 @@ Dvision_model.vision._en.stages.1.downsample.proj.1.reparam_conv.bias conv2d_12 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1163, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 629, 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 input = module(input) @@ -652,53 +652,53 @@ Gvision_model.vision._en.stages.1.blocks.0.token_mixer.reparam_conv.bias conv2d 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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 - x = self.blocks(x) - File "/home/batman/xx/.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 - x = self.token_mixer(x) - File "/home/batman/xx/.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 - 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* -group* -pads@@@@* -auto_pad"NOTSET* -strides@@* - dilations@@J - 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.conv: timm.layers.conv_bn_act.ConvNormAct/vision_model.vision._en.stages.1.blocks.0.mlp.conv.bn: timm.layers.norm_act.BatchNormAct2d/_native_batch_norm_legit_no_training_2: aten._native_batch_norm_legit_no_training.defaultJV -!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.01, 0.001), 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 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 191, in forward - return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + x = self.token_mixer(x) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 718, 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 + 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_287"Conv* +group* +pads@@@@* +auto_pad"NOTSET* +strides@@* + dilations@@J + 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.conv: timm.layers.conv_bn_act.ConvNormAct/vision_model.vision._en.stages.1.blocks.0.mlp.conv.bn: xx.training.path.allnorm.AllNormAct2d/_native_batch_norm_legit_no_training_2: aten._native_batch_norm_legit_no_training.defaultJV +!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', 'xx.training.path.allnorm.AllNormAct2d', '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, %repeat_10, %repeat_11, %repeat_8, %repeat_9, 0.01, 0.001), 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 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 196, in forward + return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, in forward + x = self.forward_features(x) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 + input = module(input) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, 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 x = self.bn(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward - x = F.batch_norm( + File "/home/batman/xx/xx/training/path/allnorm.py", line 60, in forward + return self.act(self.drop(self._allnorm_forward(x)))  getitem_6 8vision_model.vision._en.stages.1.blocks.0.mlp.fc1.weight @@ -714,17 +714,17 @@ Cvision_model.vision._en.stages.1.blocks.0.mlp.conv.conv.weight_bias getitem_6 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/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, 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 return self._conv_forward(input, self.weight, self.bias) @@ -737,17 +737,17 @@ Cvision_model.vision._en.stages.1.blocks.0.mlp.conv.conv.weight_bias getitem_6 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 817, in forward x = self.act(x) File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') @@ -766,17 +766,17 @@ Cvision_model.vision._en.stages.1.blocks.0.mlp.conv.conv.weight_bias getitem_6 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/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 819, 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 return self._conv_forward(input, self.weight, self.bias) @@ -790,17 +790,17 @@ node_mul_2"MulJ 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 647, in forward return x.mul_(self.gamma) if self.inplace else x * self.gamma  conv2d_13 @@ -812,15 +812,15 @@ node_add_2"AddJ 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x)))  add_2 @@ -837,53 +837,53 @@ Gvision_model.vision._en.stages.1.blocks.1.token_mixer.reparam_conv.bias conv2d 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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 - x = self.blocks(x) - File "/home/batman/xx/.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 - x = self.token_mixer(x) - File "/home/batman/xx/.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 - 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* -group* -pads@@@@* -auto_pad"NOTSET* -strides@@* - dilations@@J - 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.conv: timm.layers.conv_bn_act.ConvNormAct/vision_model.vision._en.stages.1.blocks.1.mlp.conv.bn: timm.layers.norm_act.BatchNormAct2d/_native_batch_norm_legit_no_training_3: aten._native_batch_norm_legit_no_training.defaultJV -!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.01, 0.001), 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 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 191, in forward - return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + x = self.token_mixer(x) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 718, 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 + 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_289"Conv* +group* +pads@@@@* +auto_pad"NOTSET* +strides@@* + dilations@@J + 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.conv: timm.layers.conv_bn_act.ConvNormAct/vision_model.vision._en.stages.1.blocks.1.mlp.conv.bn: xx.training.path.allnorm.AllNormAct2d/_native_batch_norm_legit_no_training_3: aten._native_batch_norm_legit_no_training.defaultJV +!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', 'xx.training.path.allnorm.AllNormAct2d', '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, %repeat_14, %repeat_15, %repeat_12, %repeat_13, 0.01, 0.001), 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 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 196, in forward + return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, in forward + x = self.forward_features(x) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 + input = module(input) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, 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 x = self.bn(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward - x = F.batch_norm( + File "/home/batman/xx/xx/training/path/allnorm.py", line 60, in forward + return self.act(self.drop(self._allnorm_forward(x)))  getitem_9 8vision_model.vision._en.stages.1.blocks.1.mlp.fc1.weight @@ -899,17 +899,17 @@ Cvision_model.vision._en.stages.1.blocks.1.mlp.conv.conv.weight_bias getitem_9 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/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, 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 return self._conv_forward(input, self.weight, self.bias) @@ -922,17 +922,17 @@ Cvision_model.vision._en.stages.1.blocks.1.mlp.conv.conv.weight_bias getitem_9 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 817, in forward x = self.act(x) File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') @@ -951,17 +951,17 @@ Cvision_model.vision._en.stages.1.blocks.1.mlp.conv.conv.weight_bias getitem_9 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/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 819, 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 return self._conv_forward(input, self.weight, self.bias) @@ -975,17 +975,17 @@ node_mul_3"MulJ 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 647, in forward return x.mul_(self.gamma) if self.inplace else x * self.gamma  conv2d_17 @@ -997,15 +997,15 @@ node_add_3"AddJ 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x)))  add_3 @@ -1022,17 +1022,17 @@ Dvision_model.vision._en.stages.2.downsample.proj.0.reparam_conv.bias conv2d_21 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1163, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 629, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 376, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 377, 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 return self._conv_forward(input, self.weight, self.bias) @@ -1051,13 +1051,13 @@ Dvision_model.vision._en.stages.2.downsample.proj.1.reparam_conv.bias conv2d_22 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1163, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 629, 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 input = module(input) @@ -1074,13 +1074,13 @@ Dvision_model.vision._en.stages.2.downsample.proj.1.reparam_conv.bias conv2d_22 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1163, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 629, 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 input = module(input) @@ -1103,54 +1103,54 @@ Gvision_model.vision._en.stages.2.blocks.0.token_mixer.reparam_conv.bias conv2d 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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 - x = self.blocks(x) - File "/home/batman/xx/.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 - x = self.token_mixer(x) - File "/home/batman/xx/.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 - 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 -getitem_12 node_Conv_190"Conv* -group* -pads@@@@* -auto_pad"NOTSET* -strides@@* - dilations@@J - 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.conv: timm.layers.conv_bn_act.ConvNormAct/vision_model.vision._en.stages.2.blocks.0.mlp.conv.bn: timm.layers.norm_act.BatchNormAct2d/_native_batch_norm_legit_no_training_4: aten._native_batch_norm_legit_no_training.defaultJV -!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.01, 0.001), 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 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 191, in forward - return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + x = self.token_mixer(x) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 718, 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 + 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 +getitem_12 node_Conv_291"Conv* +group* +pads@@@@* +auto_pad"NOTSET* +strides@@* + dilations@@J + 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.conv: timm.layers.conv_bn_act.ConvNormAct/vision_model.vision._en.stages.2.blocks.0.mlp.conv.bn: xx.training.path.allnorm.AllNormAct2d/_native_batch_norm_legit_no_training_4: aten._native_batch_norm_legit_no_training.defaultJV +!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', 'xx.training.path.allnorm.AllNormAct2d', '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, %repeat_18, %repeat_19, %repeat_16, %repeat_17, 0.01, 0.001), 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 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 196, in forward + return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, in forward + x = self.forward_features(x) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 + input = module(input) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, 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 x = self.bn(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward - x = F.batch_norm( + File "/home/batman/xx/xx/training/path/allnorm.py", line 60, in forward + return self.act(self.drop(self._allnorm_forward(x)))  getitem_12 @@ -1167,17 +1167,17 @@ getitem_12 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/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, 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 return self._conv_forward(input, self.weight, self.bias) @@ -1190,17 +1190,17 @@ getitem_12 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 817, in forward x = self.act(x) File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') @@ -1219,17 +1219,17 @@ getitem_12 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/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 819, 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 return self._conv_forward(input, self.weight, self.bias) @@ -1243,17 +1243,17 @@ node_mul_4"MulJ 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 647, in forward return x.mul_(self.gamma) if self.inplace else x * self.gamma  conv2d_23 @@ -1265,15 +1265,15 @@ node_add_4"AddJ 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x)))  add_4 @@ -1290,54 +1290,54 @@ Gvision_model.vision._en.stages.2.blocks.1.token_mixer.reparam_conv.bias conv2d 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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 - x = self.blocks(x) - File "/home/batman/xx/.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 - x = self.token_mixer(x) - File "/home/batman/xx/.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 - 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 -getitem_15 node_Conv_192"Conv* -group* -pads@@@@* -auto_pad"NOTSET* -strides@@* - dilations@@J - 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.conv: timm.layers.conv_bn_act.ConvNormAct/vision_model.vision._en.stages.2.blocks.1.mlp.conv.bn: timm.layers.norm_act.BatchNormAct2d/_native_batch_norm_legit_no_training_5: aten._native_batch_norm_legit_no_training.defaultJV -!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.01, 0.001), 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 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 191, in forward - return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + x = self.token_mixer(x) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 718, 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 + 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 +getitem_15 node_Conv_293"Conv* +group* +pads@@@@* +auto_pad"NOTSET* +strides@@* + dilations@@J + 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.conv: timm.layers.conv_bn_act.ConvNormAct/vision_model.vision._en.stages.2.blocks.1.mlp.conv.bn: xx.training.path.allnorm.AllNormAct2d/_native_batch_norm_legit_no_training_5: aten._native_batch_norm_legit_no_training.defaultJV +!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', 'xx.training.path.allnorm.AllNormAct2d', '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, %repeat_22, %repeat_23, %repeat_20, %repeat_21, 0.01, 0.001), 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 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 196, in forward + return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, in forward + x = self.forward_features(x) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 + input = module(input) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, 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 x = self.bn(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward - x = F.batch_norm( + File "/home/batman/xx/xx/training/path/allnorm.py", line 60, in forward + return self.act(self.drop(self._allnorm_forward(x)))  getitem_15 @@ -1354,17 +1354,17 @@ getitem_15 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/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, 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 return self._conv_forward(input, self.weight, self.bias) @@ -1377,17 +1377,17 @@ getitem_15 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 817, in forward x = self.act(x) File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') @@ -1406,17 +1406,17 @@ getitem_15 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/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 819, 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 return self._conv_forward(input, self.weight, self.bias) @@ -1430,17 +1430,17 @@ node_mul_5"MulJ 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 647, in forward return x.mul_(self.gamma) if self.inplace else x * self.gamma  conv2d_27 @@ -1452,15 +1452,15 @@ node_add_5"AddJ 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x)))  add_5 @@ -1477,54 +1477,54 @@ Gvision_model.vision._en.stages.2.blocks.2.token_mixer.reparam_conv.bias conv2d 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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 - x = self.blocks(x) - File "/home/batman/xx/.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 - x = self.token_mixer(x) - File "/home/batman/xx/.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 - 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 -getitem_18 node_Conv_194"Conv* -group* -pads@@@@* -auto_pad"NOTSET* -strides@@* - dilations@@J - 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.conv: timm.layers.conv_bn_act.ConvNormAct/vision_model.vision._en.stages.2.blocks.2.mlp.conv.bn: timm.layers.norm_act.BatchNormAct2d/_native_batch_norm_legit_no_training_6: aten._native_batch_norm_legit_no_training.defaultJV -!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.01, 0.001), 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 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 191, in forward - return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + x = self.token_mixer(x) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 718, 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 + 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 +getitem_18 node_Conv_295"Conv* +group* +pads@@@@* +auto_pad"NOTSET* +strides@@* + dilations@@J + 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.conv: timm.layers.conv_bn_act.ConvNormAct/vision_model.vision._en.stages.2.blocks.2.mlp.conv.bn: xx.training.path.allnorm.AllNormAct2d/_native_batch_norm_legit_no_training_6: aten._native_batch_norm_legit_no_training.defaultJV +!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', 'xx.training.path.allnorm.AllNormAct2d', '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, %repeat_26, %repeat_27, %repeat_24, %repeat_25, 0.01, 0.001), 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 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 196, in forward + return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, in forward + x = self.forward_features(x) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 + input = module(input) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, 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 x = self.bn(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward - x = F.batch_norm( + File "/home/batman/xx/xx/training/path/allnorm.py", line 60, in forward + return self.act(self.drop(self._allnorm_forward(x)))  getitem_18 @@ -1541,17 +1541,17 @@ getitem_18 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/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, 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 return self._conv_forward(input, self.weight, self.bias) @@ -1564,17 +1564,17 @@ getitem_18 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 817, in forward x = self.act(x) File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') @@ -1593,17 +1593,17 @@ getitem_18 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/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 819, 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 return self._conv_forward(input, self.weight, self.bias) @@ -1617,17 +1617,17 @@ node_mul_6"MulJ 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 647, in forward return x.mul_(self.gamma) if self.inplace else x * self.gamma  conv2d_31 @@ -1639,15 +1639,15 @@ node_add_6"AddJ 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x)))  add_6 @@ -1664,54 +1664,54 @@ Gvision_model.vision._en.stages.2.blocks.3.token_mixer.reparam_conv.bias conv2d 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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 - x = self.blocks(x) - File "/home/batman/xx/.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 - x = self.token_mixer(x) - File "/home/batman/xx/.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 - 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 -getitem_21 node_Conv_196"Conv* -group* -pads@@@@* -auto_pad"NOTSET* -strides@@* - dilations@@J - 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.conv: timm.layers.conv_bn_act.ConvNormAct/vision_model.vision._en.stages.2.blocks.3.mlp.conv.bn: timm.layers.norm_act.BatchNormAct2d/_native_batch_norm_legit_no_training_7: aten._native_batch_norm_legit_no_training.defaultJV -!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.01, 0.001), 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 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 191, in forward - return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + x = self.token_mixer(x) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 718, 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 + 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 +getitem_21 node_Conv_297"Conv* +group* +pads@@@@* +auto_pad"NOTSET* +strides@@* + dilations@@J + 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.conv: timm.layers.conv_bn_act.ConvNormAct/vision_model.vision._en.stages.2.blocks.3.mlp.conv.bn: xx.training.path.allnorm.AllNormAct2d/_native_batch_norm_legit_no_training_7: aten._native_batch_norm_legit_no_training.defaultJV +!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', 'xx.training.path.allnorm.AllNormAct2d', '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, %repeat_30, %repeat_31, %repeat_28, %repeat_29, 0.01, 0.001), 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 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 196, in forward + return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, in forward + x = self.forward_features(x) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 + input = module(input) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, 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 x = self.bn(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward - x = F.batch_norm( + File "/home/batman/xx/xx/training/path/allnorm.py", line 60, in forward + return self.act(self.drop(self._allnorm_forward(x)))  getitem_21 @@ -1728,17 +1728,17 @@ getitem_21 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/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, 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 return self._conv_forward(input, self.weight, self.bias) @@ -1751,17 +1751,17 @@ getitem_21 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 817, in forward x = self.act(x) File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') @@ -1780,17 +1780,17 @@ getitem_21 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/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 819, 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 return self._conv_forward(input, self.weight, self.bias) @@ -1804,17 +1804,17 @@ node_mul_7"MulJ 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 647, in forward return x.mul_(self.gamma) if self.inplace else x * self.gamma  conv2d_35 @@ -1826,15 +1826,15 @@ node_add_7"AddJ 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x)))  add_7 @@ -1851,54 +1851,54 @@ Gvision_model.vision._en.stages.2.blocks.4.token_mixer.reparam_conv.bias conv2d 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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 - x = self.blocks(x) - File "/home/batman/xx/.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 - x = self.token_mixer(x) - File "/home/batman/xx/.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 - 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 -getitem_24 node_Conv_198"Conv* -group* -pads@@@@* -auto_pad"NOTSET* -strides@@* - dilations@@J - 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.conv: timm.layers.conv_bn_act.ConvNormAct/vision_model.vision._en.stages.2.blocks.4.mlp.conv.bn: timm.layers.norm_act.BatchNormAct2d/_native_batch_norm_legit_no_training_8: aten._native_batch_norm_legit_no_training.defaultJV -!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.01, 0.001), 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 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 191, in forward - return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + x = self.token_mixer(x) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 718, 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 + 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 +getitem_24 node_Conv_299"Conv* +group* +pads@@@@* +auto_pad"NOTSET* +strides@@* + dilations@@J + 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.conv: timm.layers.conv_bn_act.ConvNormAct/vision_model.vision._en.stages.2.blocks.4.mlp.conv.bn: xx.training.path.allnorm.AllNormAct2d/_native_batch_norm_legit_no_training_8: aten._native_batch_norm_legit_no_training.defaultJV +!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', 'xx.training.path.allnorm.AllNormAct2d', '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, %repeat_34, %repeat_35, %repeat_32, %repeat_33, 0.01, 0.001), 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 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 196, in forward + return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, in forward + x = self.forward_features(x) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 + input = module(input) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, 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 x = self.bn(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward - x = F.batch_norm( + File "/home/batman/xx/xx/training/path/allnorm.py", line 60, in forward + return self.act(self.drop(self._allnorm_forward(x)))  getitem_24 @@ -1915,17 +1915,17 @@ getitem_24 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/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, 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 return self._conv_forward(input, self.weight, self.bias) @@ -1938,17 +1938,17 @@ getitem_24 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 817, in forward x = self.act(x) File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') @@ -1967,17 +1967,17 @@ getitem_24 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/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 819, 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 return self._conv_forward(input, self.weight, self.bias) @@ -1991,17 +1991,17 @@ node_mul_8"MulJ 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 647, in forward return x.mul_(self.gamma) if self.inplace else x * self.gamma  conv2d_39 @@ -2013,15 +2013,15 @@ node_add_8"AddJ 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x)))  add_8 @@ -2038,54 +2038,54 @@ Gvision_model.vision._en.stages.2.blocks.5.token_mixer.reparam_conv.bias conv2d 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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 - x = self.blocks(x) - File "/home/batman/xx/.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 - x = self.token_mixer(x) - File "/home/batman/xx/.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 - 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 -getitem_27 node_Conv_200"Conv* -group* -pads@@@@* -auto_pad"NOTSET* -strides@@* - dilations@@J - 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.conv: timm.layers.conv_bn_act.ConvNormAct/vision_model.vision._en.stages.2.blocks.5.mlp.conv.bn: timm.layers.norm_act.BatchNormAct2d/_native_batch_norm_legit_no_training_9: aten._native_batch_norm_legit_no_training.defaultJV -!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.01, 0.001), 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 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 191, in forward - return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + x = self.token_mixer(x) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 718, 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 + 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 +getitem_27 node_Conv_301"Conv* +group* +pads@@@@* +auto_pad"NOTSET* +strides@@* + dilations@@J + 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.conv: timm.layers.conv_bn_act.ConvNormAct/vision_model.vision._en.stages.2.blocks.5.mlp.conv.bn: xx.training.path.allnorm.AllNormAct2d/_native_batch_norm_legit_no_training_9: aten._native_batch_norm_legit_no_training.defaultJV +!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', 'xx.training.path.allnorm.AllNormAct2d', '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, %repeat_38, %repeat_39, %repeat_36, %repeat_37, 0.01, 0.001), 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 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 196, in forward + return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, in forward + x = self.forward_features(x) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 + input = module(input) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, 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 x = self.bn(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward - x = F.batch_norm( + File "/home/batman/xx/xx/training/path/allnorm.py", line 60, in forward + return self.act(self.drop(self._allnorm_forward(x)))  getitem_27 @@ -2102,17 +2102,17 @@ getitem_27 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/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, 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 return self._conv_forward(input, self.weight, self.bias) @@ -2125,17 +2125,17 @@ getitem_27 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 817, in forward x = self.act(x) File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') @@ -2154,17 +2154,17 @@ getitem_27 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/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 819, 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 return self._conv_forward(input, self.weight, self.bias) @@ -2178,17 +2178,17 @@ node_mul_9"MulJ 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 647, in forward return x.mul_(self.gamma) if self.inplace else x * self.gamma  conv2d_43 @@ -2200,15 +2200,15 @@ node_add_9"AddJ 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x)))  add_9 @@ -2225,17 +2225,17 @@ Dvision_model.vision._en.stages.3.downsample.proj.0.reparam_conv.bias conv2d_47 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1163, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 629, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 376, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 377, 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 return self._conv_forward(input, self.weight, self.bias) @@ -2254,13 +2254,13 @@ Dvision_model.vision._en.stages.3.downsample.proj.1.reparam_conv.bias conv2d_48 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1163, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 629, 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 input = module(input) @@ -2277,13 +2277,13 @@ Dvision_model.vision._en.stages.3.downsample.proj.1.reparam_conv.bias conv2d_48 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1163, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 629, 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 input = module(input) @@ -2306,54 +2306,54 @@ Gvision_model.vision._en.stages.3.blocks.0.token_mixer.reparam_conv.bias conv2d 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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 - x = self.blocks(x) - File "/home/batman/xx/.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 - x = self.token_mixer(x) - File "/home/batman/xx/.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 - 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 -getitem_30 node_Conv_202"Conv* -group* -pads@@@@* -auto_pad"NOTSET* -strides@@* - dilations@@J - 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.conv: timm.layers.conv_bn_act.ConvNormAct/vision_model.vision._en.stages.3.blocks.0.mlp.conv.bn: timm.layers.norm_act.BatchNormAct2d/_native_batch_norm_legit_no_training_10: aten._native_batch_norm_legit_no_training.defaultJV -!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.01, 0.001), 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 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 191, in forward - return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + x = self.token_mixer(x) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 718, 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 + 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 +getitem_30 node_Conv_303"Conv* +group* +pads@@@@* +auto_pad"NOTSET* +strides@@* + dilations@@J + 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.conv: timm.layers.conv_bn_act.ConvNormAct/vision_model.vision._en.stages.3.blocks.0.mlp.conv.bn: xx.training.path.allnorm.AllNormAct2d/_native_batch_norm_legit_no_training_10: aten._native_batch_norm_legit_no_training.defaultJV +!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', 'xx.training.path.allnorm.AllNormAct2d', '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, %repeat_42, %repeat_43, %repeat_40, %repeat_41, 0.01, 0.001), 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 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 196, in forward + return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, in forward + x = self.forward_features(x) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 + input = module(input) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, 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 x = self.bn(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward - x = F.batch_norm( + File "/home/batman/xx/xx/training/path/allnorm.py", line 60, in forward + return self.act(self.drop(self._allnorm_forward(x)))  getitem_30 @@ -2370,17 +2370,17 @@ getitem_30 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/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, 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 return self._conv_forward(input, self.weight, self.bias) @@ -2393,17 +2393,17 @@ getitem_30 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 817, in forward x = self.act(x) File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') @@ -2422,17 +2422,17 @@ getitem_30 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/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 819, 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 return self._conv_forward(input, self.weight, self.bias) @@ -2445,17 +2445,17 @@ getitem_30 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 647, in forward return x.mul_(self.gamma) if self.inplace else x * self.gamma  conv2d_49 @@ -2466,15 +2466,15 @@ getitem_30 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x)))  add_10 @@ -2491,54 +2491,54 @@ Gvision_model.vision._en.stages.3.blocks.1.token_mixer.reparam_conv.bias conv2d 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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 - x = self.blocks(x) - File "/home/batman/xx/.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 - x = self.token_mixer(x) - File "/home/batman/xx/.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 - 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 -getitem_33 node_Conv_204"Conv* -group* -pads@@@@* -auto_pad"NOTSET* -strides@@* - dilations@@J - 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.conv: timm.layers.conv_bn_act.ConvNormAct/vision_model.vision._en.stages.3.blocks.1.mlp.conv.bn: timm.layers.norm_act.BatchNormAct2d/_native_batch_norm_legit_no_training_11: aten._native_batch_norm_legit_no_training.defaultJV -!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.01, 0.001), 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 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 191, in forward - return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + x = self.token_mixer(x) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 718, 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 + 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 +getitem_33 node_Conv_305"Conv* +group* +pads@@@@* +auto_pad"NOTSET* +strides@@* + dilations@@J + 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.conv: timm.layers.conv_bn_act.ConvNormAct/vision_model.vision._en.stages.3.blocks.1.mlp.conv.bn: xx.training.path.allnorm.AllNormAct2d/_native_batch_norm_legit_no_training_11: aten._native_batch_norm_legit_no_training.defaultJV +!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', 'xx.training.path.allnorm.AllNormAct2d', '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, %repeat_46, %repeat_47, %repeat_44, %repeat_45, 0.01, 0.001), 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 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 196, in forward + return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, in forward + x = self.forward_features(x) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 + input = module(input) + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, 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 x = self.bn(x) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/norm_act.py", line 136, in forward - x = F.batch_norm( + File "/home/batman/xx/xx/training/path/allnorm.py", line 60, in forward + return self.act(self.drop(self._allnorm_forward(x)))  getitem_33 @@ -2555,17 +2555,17 @@ getitem_33 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/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 816, 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 return self._conv_forward(input, self.weight, self.bias) @@ -2578,17 +2578,17 @@ getitem_33 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_trace File "/home/batman/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 817, in forward x = self.act(x) File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/layers/activations.py", line 159, in forward return F.gelu(input, approximate='tanh') @@ -2607,17 +2607,17 @@ getitem_33 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/xx/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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 819, 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 return self._conv_forward(input, self.weight, self.bias) @@ -2630,17 +2630,17 @@ getitem_33 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 647, in forward return x.mul_(self.gamma) if self.inplace else x * self.gamma  conv2d_53 @@ -2651,15 +2651,15 @@ getitem_33 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1168, 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 input = module(input) - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1002, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1003, in forward x = x + self.drop_path(self.layer_scale(self.mlp(x)))  add_11 @@ -2676,9 +2676,9 @@ getitem_33 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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 return self.act(self.se(self.reparam_conv(x))) @@ -2686,7 +2686,7 @@ getitem_33 return self._conv_forward(input, self.weight, self.bias) conv2d_57 -val_137mean node_mean" +val_190mean node_mean" ReduceMean* noop_with_empty_axes* keepdimsJ @@ -2696,9 +2696,9 @@ ReduceMean* 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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 return self.act(self.se(self.reparam_conv(x))) @@ -2719,9 +2719,9 @@ ReduceMean* 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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 return self.act(self.se(self.reparam_conv(x))) @@ -2737,9 +2737,9 @@ ReduceMean* 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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 return self.act(self.se(self.reparam_conv(x))) @@ -2762,9 +2762,9 @@ ReduceMean* 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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 return self.act(self.se(self.reparam_conv(x))) @@ -2780,9 +2780,9 @@ ReduceMean* 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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 return self.act(self.se(self.reparam_conv(x))) @@ -2799,9 +2799,9 @@ ReduceMean* 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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 return self.act(self.se(self.reparam_conv(x))) @@ -2816,9 +2816,9 @@ ReduceMean* 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1418, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1419, 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 return self.act(self.se(self.reparam_conv(x))) @@ -2826,7 +2826,7 @@ ReduceMean* return F.gelu(input, approximate='tanh')  gelu_18 -val_140mean_1 node_mean_1" +val_193mean_1 node_mean_1" ReduceMean* noop_with_empty_axes* keepdimsJ @@ -2836,9 +2836,9 @@ ReduceMean* 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1421, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1422, 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 x = self.global_pool(x) @@ -2848,7 +2848,7 @@ ReduceMean* return F.adaptive_avg_pool2d(input, self.output_size)  mean_1 -val_144view node_view"Reshape* +val_197view 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 @@ -2856,9 +2856,9 @@ ReduceMean* 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1421, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1422, 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 x = self.global_pool(x) @@ -2880,9 +2880,9 @@ $vision_model.vision._en.head.fc.biaslinear node_linear"Gemm* 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 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 191, in forward + File "/home/batman/xx/xx/training/path/supercombo.py", line 196, in forward return {ModelInputs.FEATURES: self._en((x - self._mean) / self._std)} - File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1421, in forward + File "/home/batman/xx/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 1422, 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 x = self.fc(x) @@ -55553,7 +55553,7 @@ BIvision_model.point_policy.hydra.final_layer.wide_from_device_euler.weightJ AC>AB@$E@DX@ݳfC@B@?mA@&TAѹ94DA*[ BGvision_model.point_policy.hydra.final_layer.wide_from_device_euler.biasJ ǯ"8*W Bvision_model.vision._meanJ0WWWWWWWWWWWWWWWWWWWWWWWW*V -Bvision_model.vision._stdJ0SSSSSSSSSSSSSSSSSSSSSSSS*Bval_137J*Bval_140J*Bval_144J*@ +Bvision_model.vision._stdJ0SSSSSSSSSSSSSSSSSSSSSSSS*Bval_190J*Bval_193J*Bval_197J*@ BCvision_model.vision._en.stages.0.blocks.0.mlp.conv.conv.weight_biasJ4444444444444444444444444444444444444444444444444444444444444444*@ BCvision_model.vision._en.stages.0.blocks.1.mlp.conv.conv.weight_biasJ]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]* BCvision_model.vision._en.stages.1.blocks.0.mlp.conv.conv.weight_biasJ * @@ -57102,21 +57102,21 @@ Gvision_model.point_policy.hydra.final_layer.wide_from_device_euler.bias /pkg.torch.export.graph_signature.InputSpec.kindBUFFER"= 5pkg.torch.export.graph_signature.InputSpec.persistentTrue"? !pkg.torch.onnx.original_node_nameb_vision_model_vision__stdjU -val_137 +val_190  "> -$pkg.onnxscript.optimizer.folded_from['val_135', 'val_136']jU -val_140 +$pkg.onnxscript.optimizer.folded_from['val_188', 'val_189']jU +val_193  "> -$pkg.onnxscript.optimizer.folded_from['val_138', 'val_139']jU -val_144 +$pkg.onnxscript.optimizer.folded_from['val_191', 'val_192']jU +val_197  "> -$pkg.onnxscript.optimizer.folded_from['val_142', 'val_143']jQ +$pkg.onnxscript.optimizer.folded_from['val_195', 'val_196']jQ Cvision_model.vision._en.stages.0.blocks.0.mlp.conv.conv.weight_bias  @@ -58624,5 +58624,5 @@ cat_1: USER_OUTPUT cmVfcHJlZJRoBEs3S1dOh5RSlIwEcG9zZZRoBEtXS2NOh5RSlIwWd2lkZV9mcm9tX2RldmljZV9l dWxlcpRoBEtjS2lOh5RSlIwOcm9hZF90cmFuc2Zvcm2UaARLaUt1ToeUUpSMDGhpZGRlbl9zdGF0 ZZRoBEt1TXUCToeUUpSMA3BhZJRoBEr9////Tk6HlFKUdS4= -r< -model_checkpoint(13f3a79f-126c-4603-a1b1-6c3b1f9f6c68/100 \ No newline at end of file +r8 +model_checkpoint$8c97c159-21d3-4088-8507-7e7cf721998e \ No newline at end of file