diff --git a/uncompiledmodels/driving_off_policy.onnx b/uncompiledmodels/driving_off_policy.onnx index 9809ff165..edbdfaba9 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 f52e6e2aa..11c153428 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 0e9012654..d4fef73c0 100644 --- a/uncompiledmodels/driving_vision.onnx +++ b/uncompiledmodels/driving_vision.onnx @@ -1,5 +1,5 @@  -pytorch 2.11.0+cu130:Ю +pytorch 2.12.0+cu130:Ю  img_to_copy node__to_copy"Cast* @@ -80,7 +80,7 @@ _to_copy_1catnode_cat"Concat* input = module(input) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward return self.act(self.se(self.reparam_conv(x))) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2dgelu node_gelu"Gelu* @@ -124,7 +124,7 @@ _to_copy_1catnode_cat"Concat* input = module(input) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward return self.act(self.se(self.reparam_conv(x))) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_1gelu_1 node_gelu_1"Gelu* @@ -168,7 +168,7 @@ _to_copy_1catnode_cat"Concat* input = module(input) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward return self.act(self.se(self.reparam_conv(x))) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_2gelu_2 node_gelu_2"Gelu* @@ -216,7 +216,7 @@ Gvision_model.vision._en.stages.0.blocks.0.token_mixer.reparam_conv.biasconv2d x = self.token_mixer(x) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward x = self.reparam_conv(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_3 @@ -279,7 +279,7 @@ File "/home/batman/xx3/./ml_tools/openpilot_compile/compile_supercombo.py", line x = x + self.drop_path(self.layer_scale(self.mlp(x))) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward x = self.fc1(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_5gelu_3 node_gelu_3"Gelu* @@ -331,7 +331,7 @@ File "/home/batman/xx3/./ml_tools/openpilot_compile/compile_supercombo.py", line x = x + self.drop_path(self.layer_scale(self.mlp(x))) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward x = self.fc2(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_6 @@ -400,7 +400,7 @@ Gvision_model.vision._en.stages.0.blocks.1.token_mixer.reparam_conv.biasconv2d x = self.token_mixer(x) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward x = self.reparam_conv(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_7 @@ -463,7 +463,7 @@ File "/home/batman/xx3/./ml_tools/openpilot_compile/compile_supercombo.py", line x = x + self.drop_path(self.layer_scale(self.mlp(x))) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward x = self.fc1(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_9gelu_4 node_gelu_4"Gelu* @@ -515,7 +515,7 @@ File "/home/batman/xx3/./ml_tools/openpilot_compile/compile_supercombo.py", line x = x + self.drop_path(self.layer_scale(self.mlp(x))) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward x = self.fc2(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_10 @@ -586,7 +586,7 @@ Dvision_model.vision._en.stages.1.downsample.proj.0.reparam_conv.bias conv2d_11 input = module(input) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 376, in forward out = self.reparam_conv(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_11 @@ -615,7 +615,7 @@ Dvision_model.vision._en.stages.1.downsample.proj.1.reparam_conv.bias conv2d_12 input = module(input) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward return self.act(self.se(self.reparam_conv(x))) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_12gelu_5 node_gelu_5"Gelu* @@ -667,7 +667,7 @@ Gvision_model.vision._en.stages.1.blocks.0.token_mixer.reparam_conv.bias conv2d x = self.token_mixer(x) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward x = self.reparam_conv(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_13 @@ -730,7 +730,7 @@ File "/home/batman/xx3/./ml_tools/openpilot_compile/compile_supercombo.py", line x = x + self.drop_path(self.layer_scale(self.mlp(x))) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward x = self.fc1(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_15gelu_6 node_gelu_6"Gelu* @@ -782,7 +782,7 @@ File "/home/batman/xx3/./ml_tools/openpilot_compile/compile_supercombo.py", line x = x + self.drop_path(self.layer_scale(self.mlp(x))) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward x = self.fc2(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_16 @@ -853,7 +853,7 @@ Gvision_model.vision._en.stages.1.blocks.1.token_mixer.reparam_conv.bias conv2d x = self.token_mixer(x) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward x = self.reparam_conv(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_17 @@ -916,7 +916,7 @@ File "/home/batman/xx3/./ml_tools/openpilot_compile/compile_supercombo.py", line x = x + self.drop_path(self.layer_scale(self.mlp(x))) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward x = self.fc1(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_19gelu_7 node_gelu_7"Gelu* @@ -968,7 +968,7 @@ File "/home/batman/xx3/./ml_tools/openpilot_compile/compile_supercombo.py", line x = x + self.drop_path(self.layer_scale(self.mlp(x))) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward x = self.fc2(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_20 @@ -1039,7 +1039,7 @@ Dvision_model.vision._en.stages.2.downsample.proj.0.reparam_conv.bias conv2d_21 input = module(input) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 376, in forward out = self.reparam_conv(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_21 @@ -1068,7 +1068,7 @@ Dvision_model.vision._en.stages.2.downsample.proj.1.reparam_conv.bias conv2d_22 input = module(input) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward return self.act(self.se(self.reparam_conv(x))) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_22gelu_8 node_gelu_8"Gelu* @@ -1120,7 +1120,7 @@ Gvision_model.vision._en.stages.2.blocks.0.token_mixer.reparam_conv.bias conv2d x = self.token_mixer(x) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward x = self.reparam_conv(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_23 @@ -1185,7 +1185,7 @@ getitem_12 x = x + self.drop_path(self.layer_scale(self.mlp(x))) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward x = self.fc1(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_25gelu_9 node_gelu_9"Gelu* @@ -1237,7 +1237,7 @@ getitem_12 x = x + self.drop_path(self.layer_scale(self.mlp(x))) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward x = self.fc2(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_26 @@ -1308,7 +1308,7 @@ Gvision_model.vision._en.stages.2.blocks.1.token_mixer.reparam_conv.bias conv2d x = self.token_mixer(x) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward x = self.reparam_conv(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_27 @@ -1373,7 +1373,7 @@ getitem_15 x = x + self.drop_path(self.layer_scale(self.mlp(x))) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward x = self.fc1(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_29gelu_10 node_gelu_10"Gelu* @@ -1425,7 +1425,7 @@ getitem_15 x = x + self.drop_path(self.layer_scale(self.mlp(x))) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward x = self.fc2(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_30 @@ -1496,7 +1496,7 @@ Gvision_model.vision._en.stages.2.blocks.2.token_mixer.reparam_conv.bias conv2d x = self.token_mixer(x) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward x = self.reparam_conv(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_31 @@ -1561,7 +1561,7 @@ getitem_18 x = x + self.drop_path(self.layer_scale(self.mlp(x))) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward x = self.fc1(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_33gelu_11 node_gelu_11"Gelu* @@ -1613,7 +1613,7 @@ getitem_18 x = x + self.drop_path(self.layer_scale(self.mlp(x))) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward x = self.fc2(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_34 @@ -1684,7 +1684,7 @@ Gvision_model.vision._en.stages.2.blocks.3.token_mixer.reparam_conv.bias conv2d x = self.token_mixer(x) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward x = self.reparam_conv(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_35 @@ -1749,7 +1749,7 @@ getitem_21 x = x + self.drop_path(self.layer_scale(self.mlp(x))) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward x = self.fc1(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_37gelu_12 node_gelu_12"Gelu* @@ -1801,7 +1801,7 @@ getitem_21 x = x + self.drop_path(self.layer_scale(self.mlp(x))) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward x = self.fc2(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_38 @@ -1872,7 +1872,7 @@ Gvision_model.vision._en.stages.2.blocks.4.token_mixer.reparam_conv.bias conv2d x = self.token_mixer(x) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward x = self.reparam_conv(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_39 @@ -1937,7 +1937,7 @@ getitem_24 x = x + self.drop_path(self.layer_scale(self.mlp(x))) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward x = self.fc1(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_41gelu_13 node_gelu_13"Gelu* @@ -1989,7 +1989,7 @@ getitem_24 x = x + self.drop_path(self.layer_scale(self.mlp(x))) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward x = self.fc2(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_42 @@ -2060,7 +2060,7 @@ Gvision_model.vision._en.stages.2.blocks.5.token_mixer.reparam_conv.bias conv2d x = self.token_mixer(x) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward x = self.reparam_conv(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_43 @@ -2125,7 +2125,7 @@ getitem_27 x = x + self.drop_path(self.layer_scale(self.mlp(x))) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward x = self.fc1(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_45gelu_14 node_gelu_14"Gelu* @@ -2177,7 +2177,7 @@ getitem_27 x = x + self.drop_path(self.layer_scale(self.mlp(x))) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward x = self.fc2(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_46 @@ -2248,7 +2248,7 @@ Dvision_model.vision._en.stages.3.downsample.proj.0.reparam_conv.bias conv2d_47 input = module(input) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 376, in forward out = self.reparam_conv(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_47 @@ -2277,7 +2277,7 @@ Dvision_model.vision._en.stages.3.downsample.proj.1.reparam_conv.bias conv2d_48 input = module(input) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward return self.act(self.se(self.reparam_conv(x))) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_48gelu_15 node_gelu_15"Gelu* @@ -2329,7 +2329,7 @@ Gvision_model.vision._en.stages.3.blocks.0.token_mixer.reparam_conv.bias conv2d x = self.token_mixer(x) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward x = self.reparam_conv(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_49 @@ -2394,7 +2394,7 @@ getitem_30 x = x + self.drop_path(self.layer_scale(self.mlp(x))) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward x = self.fc1(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_51gelu_16 node_gelu_16"Gelu* @@ -2446,7 +2446,7 @@ getitem_30 x = x + self.drop_path(self.layer_scale(self.mlp(x))) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward x = self.fc2(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_52 @@ -2515,7 +2515,7 @@ Gvision_model.vision._en.stages.3.blocks.1.token_mixer.reparam_conv.bias conv2d x = self.token_mixer(x) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 717, in forward x = self.reparam_conv(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_53 @@ -2580,7 +2580,7 @@ getitem_33 x = x + self.drop_path(self.layer_scale(self.mlp(x))) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 815, in forward x = self.fc1(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_55gelu_17 node_gelu_17"Gelu* @@ -2632,7 +2632,7 @@ getitem_33 x = x + self.drop_path(self.layer_scale(self.mlp(x))) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 818, in forward x = self.fc2(x) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_56 @@ -2695,7 +2695,7 @@ getitem_33 x = self.forward_features(x) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/models/fastvit.py", line 155, in forward return self.act(self.se(self.reparam_conv(x))) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias) conv2d_57 @@ -2740,7 +2740,7 @@ ReduceMean* return self.act(self.se(self.reparam_conv(x))) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/squeeze_excite.py", line 60, in forward x_se = self.fc1(x_se) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_58relu node_relu"ReluJ @@ -2783,7 +2783,7 @@ ReduceMean* return self.act(self.se(self.reparam_conv(x))) File "/home/batman/xx3/.venv/lib/python3.12/site-packages/timm/layers/squeeze_excite.py", line 62, in forward x_se = self.fc2(x_se) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 553, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 565, in forward return self._conv_forward(input, self.weight, self.bias)  conv2d_59sigmoid node_sigmoid"SigmoidJ @@ -2922,7 +2922,7 @@ stash_type x = self.mlp1(x) + x File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/normalization.py", line 229, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/normalization.py", line 228, in forward return F.layer_norm(  @@ -3017,7 +3017,7 @@ stash_type x = self.mlp2(x) + x File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/normalization.py", line 229, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/normalization.py", line 228, in forward return F.layer_norm(  layer_norm_1 @@ -3111,7 +3111,7 @@ stash_type k: self.head_mlp[k](in_feats) + in_feats if k in self.head_mlp else in_feats File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/container.py", line 253, in forward input = module(input) - File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/normalization.py", line 229, in forward + File "/home/batman/xx3/.venv/lib/python3.12/site-packages/torch/nn/modules/normalization.py", line 228, in forward return F.layer_norm(  layer_norm_2