Fix model weight paths, unify tensor loading (plain/JIT), and enable robust C++/Python comparison for all model components

3 weeks ago · 02923684ba
2516 changed files with 425 additions and 97 deletions
--- a/backbone_pure_tensors/bn1_bias.pt
+++ b/backbone_pure_tensors/bn1_bias.pt
--- a/backbone_pure_tensors/bn1_num_batches_tracked.pt
+++ b/backbone_pure_tensors/bn1_num_batches_tracked.pt
--- a/backbone_pure_tensors/bn1_running_mean.pt
+++ b/backbone_pure_tensors/bn1_running_mean.pt
--- a/backbone_pure_tensors/bn1_running_var.pt
+++ b/backbone_pure_tensors/bn1_running_var.pt
--- a/backbone_pure_tensors/bn1_weight.pt
+++ b/backbone_pure_tensors/bn1_weight.pt
--- a/backbone_pure_tensors/conv1_weight.pt
+++ b/backbone_pure_tensors/conv1_weight.pt
--- a/backbone_pure_tensors/conv1_weight_copy.pt
+++ b/backbone_pure_tensors/conv1_weight_copy.pt
--- a/backbone_pure_tensors/fc_bias.pt
+++ b/backbone_pure_tensors/fc_bias.pt
--- a/backbone_pure_tensors/fc_weight.pt
+++ b/backbone_pure_tensors/fc_weight.pt
--- a/backbone_pure_tensors/layer1_0_bn1_bias.pt
+++ b/backbone_pure_tensors/layer1_0_bn1_bias.pt
--- a/backbone_pure_tensors/layer1_0_bn1_num_batches_tracked.pt
+++ b/backbone_pure_tensors/layer1_0_bn1_num_batches_tracked.pt
--- a/backbone_pure_tensors/layer1_0_bn1_running_mean.pt
+++ b/backbone_pure_tensors/layer1_0_bn1_running_mean.pt
--- a/backbone_pure_tensors/layer1_0_bn1_running_var.pt
+++ b/backbone_pure_tensors/layer1_0_bn1_running_var.pt
--- a/backbone_pure_tensors/layer1_0_bn1_weight.pt
+++ b/backbone_pure_tensors/layer1_0_bn1_weight.pt
--- a/backbone_pure_tensors/layer1_0_bn2_bias.pt
+++ b/backbone_pure_tensors/layer1_0_bn2_bias.pt
--- a/backbone_pure_tensors/layer1_0_bn2_num_batches_tracked.pt
+++ b/backbone_pure_tensors/layer1_0_bn2_num_batches_tracked.pt
--- a/backbone_pure_tensors/layer1_0_bn2_running_mean.pt
+++ b/backbone_pure_tensors/layer1_0_bn2_running_mean.pt
--- a/backbone_pure_tensors/layer1_0_bn2_running_var.pt
+++ b/backbone_pure_tensors/layer1_0_bn2_running_var.pt
--- a/backbone_pure_tensors/layer1_0_bn2_weight.pt
+++ b/backbone_pure_tensors/layer1_0_bn2_weight.pt
--- a/backbone_pure_tensors/layer1_0_bn3_bias.pt
+++ b/backbone_pure_tensors/layer1_0_bn3_bias.pt
--- a/backbone_pure_tensors/layer1_0_bn3_num_batches_tracked.pt
+++ b/backbone_pure_tensors/layer1_0_bn3_num_batches_tracked.pt
--- a/backbone_pure_tensors/layer1_0_bn3_running_mean.pt
+++ b/backbone_pure_tensors/layer1_0_bn3_running_mean.pt
--- a/backbone_pure_tensors/layer1_0_bn3_running_var.pt
+++ b/backbone_pure_tensors/layer1_0_bn3_running_var.pt
--- a/backbone_pure_tensors/layer1_0_bn3_weight.pt
+++ b/backbone_pure_tensors/layer1_0_bn3_weight.pt
--- a/backbone_pure_tensors/layer1_0_conv1_weight.pt
+++ b/backbone_pure_tensors/layer1_0_conv1_weight.pt
--- a/backbone_pure_tensors/layer1_0_conv2_weight.pt
+++ b/backbone_pure_tensors/layer1_0_conv2_weight.pt
--- a/backbone_pure_tensors/layer1_0_conv3_weight.pt
+++ b/backbone_pure_tensors/layer1_0_conv3_weight.pt
--- a/backbone_pure_tensors/layer1_0_downsample_0_weight.pt
+++ b/backbone_pure_tensors/layer1_0_downsample_0_weight.pt
--- a/backbone_pure_tensors/layer1_0_downsample_1_bias.pt
+++ b/backbone_pure_tensors/layer1_0_downsample_1_bias.pt
--- a/backbone_pure_tensors/layer1_0_downsample_1_num_batches_tracked.pt
+++ b/backbone_pure_tensors/layer1_0_downsample_1_num_batches_tracked.pt
--- a/backbone_pure_tensors/layer1_0_downsample_1_running_mean.pt
+++ b/backbone_pure_tensors/layer1_0_downsample_1_running_mean.pt
--- a/backbone_pure_tensors/layer1_0_downsample_1_running_var.pt
+++ b/backbone_pure_tensors/layer1_0_downsample_1_running_var.pt
--- a/backbone_pure_tensors/layer1_0_downsample_1_weight.pt
+++ b/backbone_pure_tensors/layer1_0_downsample_1_weight.pt
--- a/backbone_pure_tensors/layer1_1_bn1_bias.pt
+++ b/backbone_pure_tensors/layer1_1_bn1_bias.pt
--- a/backbone_pure_tensors/layer1_1_bn1_num_batches_tracked.pt
+++ b/backbone_pure_tensors/layer1_1_bn1_num_batches_tracked.pt
--- a/backbone_pure_tensors/layer1_1_bn1_running_mean.pt
+++ b/backbone_pure_tensors/layer1_1_bn1_running_mean.pt
--- a/backbone_pure_tensors/layer1_1_bn1_running_var.pt
+++ b/backbone_pure_tensors/layer1_1_bn1_running_var.pt
--- a/backbone_pure_tensors/layer1_1_bn1_weight.pt
+++ b/backbone_pure_tensors/layer1_1_bn1_weight.pt
--- a/backbone_pure_tensors/layer1_1_bn2_bias.pt
+++ b/backbone_pure_tensors/layer1_1_bn2_bias.pt
--- a/backbone_pure_tensors/layer1_1_bn2_num_batches_tracked.pt
+++ b/backbone_pure_tensors/layer1_1_bn2_num_batches_tracked.pt
--- a/backbone_pure_tensors/layer1_1_bn2_running_mean.pt
+++ b/backbone_pure_tensors/layer1_1_bn2_running_mean.pt
--- a/backbone_pure_tensors/layer1_1_bn2_running_var.pt
+++ b/backbone_pure_tensors/layer1_1_bn2_running_var.pt
--- a/backbone_pure_tensors/layer1_1_bn2_weight.pt
+++ b/backbone_pure_tensors/layer1_1_bn2_weight.pt
--- a/backbone_pure_tensors/layer1_1_bn3_bias.pt
+++ b/backbone_pure_tensors/layer1_1_bn3_bias.pt
--- a/backbone_pure_tensors/layer1_1_bn3_num_batches_tracked.pt
+++ b/backbone_pure_tensors/layer1_1_bn3_num_batches_tracked.pt
--- a/backbone_pure_tensors/layer1_1_bn3_running_mean.pt
+++ b/backbone_pure_tensors/layer1_1_bn3_running_mean.pt
--- a/backbone_pure_tensors/layer1_1_bn3_running_var.pt
+++ b/backbone_pure_tensors/layer1_1_bn3_running_var.pt
--- a/backbone_pure_tensors/layer1_1_bn3_weight.pt
+++ b/backbone_pure_tensors/layer1_1_bn3_weight.pt
--- a/backbone_pure_tensors/layer1_1_conv1_weight.pt
+++ b/backbone_pure_tensors/layer1_1_conv1_weight.pt
--- a/backbone_pure_tensors/layer1_1_conv2_weight.pt
+++ b/backbone_pure_tensors/layer1_1_conv2_weight.pt
--- a/backbone_pure_tensors/layer1_1_conv3_weight.pt
+++ b/backbone_pure_tensors/layer1_1_conv3_weight.pt
--- a/backbone_pure_tensors/layer1_2_bn1_bias.pt
+++ b/backbone_pure_tensors/layer1_2_bn1_bias.pt
--- a/backbone_pure_tensors/layer1_2_bn1_num_batches_tracked.pt
+++ b/backbone_pure_tensors/layer1_2_bn1_num_batches_tracked.pt
--- a/backbone_pure_tensors/layer1_2_bn1_running_mean.pt
+++ b/backbone_pure_tensors/layer1_2_bn1_running_mean.pt
--- a/backbone_pure_tensors/layer1_2_bn1_running_var.pt
+++ b/backbone_pure_tensors/layer1_2_bn1_running_var.pt
--- a/backbone_pure_tensors/layer1_2_bn1_weight.pt
+++ b/backbone_pure_tensors/layer1_2_bn1_weight.pt
--- a/backbone_pure_tensors/layer1_2_bn2_bias.pt
+++ b/backbone_pure_tensors/layer1_2_bn2_bias.pt
--- a/backbone_pure_tensors/layer1_2_bn2_num_batches_tracked.pt
+++ b/backbone_pure_tensors/layer1_2_bn2_num_batches_tracked.pt
--- a/backbone_pure_tensors/layer1_2_bn2_running_mean.pt
+++ b/backbone_pure_tensors/layer1_2_bn2_running_mean.pt
--- a/backbone_pure_tensors/layer1_2_bn2_running_var.pt
+++ b/backbone_pure_tensors/layer1_2_bn2_running_var.pt
--- a/backbone_pure_tensors/layer1_2_bn2_weight.pt
+++ b/backbone_pure_tensors/layer1_2_bn2_weight.pt
--- a/backbone_pure_tensors/layer1_2_bn3_bias.pt
+++ b/backbone_pure_tensors/layer1_2_bn3_bias.pt
--- a/backbone_pure_tensors/layer1_2_bn3_num_batches_tracked.pt
+++ b/backbone_pure_tensors/layer1_2_bn3_num_batches_tracked.pt
--- a/backbone_pure_tensors/layer1_2_bn3_running_mean.pt
+++ b/backbone_pure_tensors/layer1_2_bn3_running_mean.pt
--- a/backbone_pure_tensors/layer1_2_bn3_running_var.pt
+++ b/backbone_pure_tensors/layer1_2_bn3_running_var.pt
--- a/backbone_pure_tensors/layer1_2_bn3_weight.pt
+++ b/backbone_pure_tensors/layer1_2_bn3_weight.pt
--- a/backbone_pure_tensors/layer1_2_conv1_weight.pt
+++ b/backbone_pure_tensors/layer1_2_conv1_weight.pt
--- a/backbone_pure_tensors/layer1_2_conv2_weight.pt
+++ b/backbone_pure_tensors/layer1_2_conv2_weight.pt
--- a/backbone_pure_tensors/layer1_2_conv3_weight.pt
+++ b/backbone_pure_tensors/layer1_2_conv3_weight.pt
--- a/backbone_pure_tensors/layer2_0_bn1_bias.pt
+++ b/backbone_pure_tensors/layer2_0_bn1_bias.pt
--- a/backbone_pure_tensors/layer2_0_bn1_num_batches_tracked.pt
+++ b/backbone_pure_tensors/layer2_0_bn1_num_batches_tracked.pt
--- a/backbone_pure_tensors/layer2_0_bn1_running_mean.pt
+++ b/backbone_pure_tensors/layer2_0_bn1_running_mean.pt
--- a/backbone_pure_tensors/layer2_0_bn1_running_var.pt
+++ b/backbone_pure_tensors/layer2_0_bn1_running_var.pt
--- a/backbone_pure_tensors/layer2_0_bn1_weight.pt
+++ b/backbone_pure_tensors/layer2_0_bn1_weight.pt
--- a/backbone_pure_tensors/layer2_0_bn2_bias.pt
+++ b/backbone_pure_tensors/layer2_0_bn2_bias.pt
--- a/backbone_pure_tensors/layer2_0_bn2_num_batches_tracked.pt
+++ b/backbone_pure_tensors/layer2_0_bn2_num_batches_tracked.pt
--- a/backbone_pure_tensors/layer2_0_bn2_running_mean.pt
+++ b/backbone_pure_tensors/layer2_0_bn2_running_mean.pt
--- a/backbone_pure_tensors/layer2_0_bn2_running_var.pt
+++ b/backbone_pure_tensors/layer2_0_bn2_running_var.pt
--- a/backbone_pure_tensors/layer2_0_bn2_weight.pt
+++ b/backbone_pure_tensors/layer2_0_bn2_weight.pt
--- a/backbone_pure_tensors/layer2_0_bn3_bias.pt
+++ b/backbone_pure_tensors/layer2_0_bn3_bias.pt
--- a/backbone_pure_tensors/layer2_0_bn3_num_batches_tracked.pt
+++ b/backbone_pure_tensors/layer2_0_bn3_num_batches_tracked.pt
--- a/backbone_pure_tensors/layer2_0_bn3_running_mean.pt
+++ b/backbone_pure_tensors/layer2_0_bn3_running_mean.pt
--- a/backbone_pure_tensors/layer2_0_bn3_running_var.pt
+++ b/backbone_pure_tensors/layer2_0_bn3_running_var.pt
--- a/backbone_pure_tensors/layer2_0_bn3_weight.pt
+++ b/backbone_pure_tensors/layer2_0_bn3_weight.pt
--- a/backbone_pure_tensors/layer2_0_conv1_weight.pt
+++ b/backbone_pure_tensors/layer2_0_conv1_weight.pt
--- a/backbone_pure_tensors/layer2_0_conv2_weight.pt
+++ b/backbone_pure_tensors/layer2_0_conv2_weight.pt
--- a/backbone_pure_tensors/layer2_0_conv3_weight.pt
+++ b/backbone_pure_tensors/layer2_0_conv3_weight.pt
--- a/backbone_pure_tensors/layer2_0_downsample_0_weight.pt
+++ b/backbone_pure_tensors/layer2_0_downsample_0_weight.pt
--- a/backbone_pure_tensors/layer2_0_downsample_1_bias.pt
+++ b/backbone_pure_tensors/layer2_0_downsample_1_bias.pt
--- a/backbone_pure_tensors/layer2_0_downsample_1_num_batches_tracked.pt
+++ b/backbone_pure_tensors/layer2_0_downsample_1_num_batches_tracked.pt
--- a/backbone_pure_tensors/layer2_0_downsample_1_running_mean.pt
+++ b/backbone_pure_tensors/layer2_0_downsample_1_running_mean.pt
--- a/backbone_pure_tensors/layer2_0_downsample_1_running_var.pt
+++ b/backbone_pure_tensors/layer2_0_downsample_1_running_var.pt
--- a/backbone_pure_tensors/layer2_0_downsample_1_weight.pt
+++ b/backbone_pure_tensors/layer2_0_downsample_1_weight.pt
--- a/backbone_pure_tensors/layer2_1_bn1_bias.pt
+++ b/backbone_pure_tensors/layer2_1_bn1_bias.pt
--- a/backbone_pure_tensors/layer2_1_bn1_num_batches_tracked.pt
+++ b/backbone_pure_tensors/layer2_1_bn1_num_batches_tracked.pt
--- a/backbone_pure_tensors/layer2_1_bn1_running_mean.pt
+++ b/backbone_pure_tensors/layer2_1_bn1_running_mean.pt
--- a/backbone_pure_tensors/layer2_1_bn1_running_var.pt
+++ b/backbone_pure_tensors/layer2_1_bn1_running_var.pt
--- a/backbone_pure_tensors/layer2_1_bn1_weight.pt
+++ b/backbone_pure_tensors/layer2_1_bn1_weight.pt
--- a/backbone_pure_tensors/layer2_1_bn2_bias.pt
+++ b/backbone_pure_tensors/layer2_1_bn2_bias.pt
--- a/backbone_pure_tensors/layer2_1_bn2_num_batches_tracked.pt
+++ b/backbone_pure_tensors/layer2_1_bn2_num_batches_tracked.pt