dollspace's picture
feat: pin DataLoader iteration fixtures v1 (#1156)
170d041 verified
{
"name": "sequential",
"batch_size": 4,
"shuffle": false,
"drop_last": false,
"seed": null,
"num_items": 10,
"feature_dim": 8,
"num_batches": 3,
"batch_sizes": [
4,
4,
2
],
"expected_num_batches": 3,
"item_features": [
[
[
0.0,
1.0,
2.0,
3.0,
4.0,
5.0,
6.0,
7.0
],
[
0.10000000149011612,
1.100000023841858,
2.0999999046325684,
3.0999999046325684,
4.099999904632568,
5.099999904632568,
6.099999904632568,
7.099999904632568
],
[
0.20000000298023224,
1.2000000476837158,
2.200000047683716,
3.200000047683716,
4.199999809265137,
5.199999809265137,
6.199999809265137,
7.199999809265137
],
[
0.30000001192092896,
1.2999999523162842,
2.299999952316284,
3.299999952316284,
4.300000190734863,
5.300000190734863,
6.300000190734863,
7.300000190734863
]
],
[
[
0.4000000059604645,
1.399999976158142,
2.4000000953674316,
3.4000000953674316,
4.400000095367432,
5.400000095367432,
6.400000095367432,
7.400000095367432
],
[
0.5,
1.5,
2.5,
3.5,
4.5,
5.5,
6.5,
7.5
],
[
0.6000000238418579,
1.600000023841858,
2.5999999046325684,
3.5999999046325684,
4.599999904632568,
5.599999904632568,
6.599999904632568,
7.599999904632568
],
[
0.699999988079071,
1.7000000476837158,
2.700000047683716,
3.700000047683716,
4.699999809265137,
5.699999809265137,
6.699999809265137,
7.699999809265137
]
],
[
[
0.800000011920929,
1.7999999523162842,
2.799999952316284,
3.799999952316284,
4.800000190734863,
5.800000190734863,
6.800000190734863,
7.800000190734863
],
[
0.8999999761581421,
1.899999976158142,
2.9000000953674316,
3.9000000953674316,
4.900000095367432,
5.900000095367432,
6.900000095367432,
7.900000095367432
]
]
],
"item_labels": [
[
0,
1,
2,
0
],
[
1,
2,
0,
1
],
[
2,
0
]
],
"full_dataset_features": [
[
0.0,
1.0,
2.0,
3.0,
4.0,
5.0,
6.0,
7.0
],
[
0.10000000149011612,
1.100000023841858,
2.0999999046325684,
3.0999999046325684,
4.099999904632568,
5.099999904632568,
6.099999904632568,
7.099999904632568
],
[
0.20000000298023224,
1.2000000476837158,
2.200000047683716,
3.200000047683716,
4.199999809265137,
5.199999809265137,
6.199999809265137,
7.199999809265137
],
[
0.30000001192092896,
1.2999999523162842,
2.299999952316284,
3.299999952316284,
4.300000190734863,
5.300000190734863,
6.300000190734863,
7.300000190734863
],
[
0.4000000059604645,
1.399999976158142,
2.4000000953674316,
3.4000000953674316,
4.400000095367432,
5.400000095367432,
6.400000095367432,
7.400000095367432
],
[
0.5,
1.5,
2.5,
3.5,
4.5,
5.5,
6.5,
7.5
],
[
0.6000000238418579,
1.600000023841858,
2.5999999046325684,
3.5999999046325684,
4.599999904632568,
5.599999904632568,
6.599999904632568,
7.599999904632568
],
[
0.699999988079071,
1.7000000476837158,
2.700000047683716,
3.700000047683716,
4.699999809265137,
5.699999809265137,
6.699999809265137,
7.699999809265137
],
[
0.800000011920929,
1.7999999523162842,
2.799999952316284,
3.799999952316284,
4.800000190734863,
5.800000190734863,
6.800000190734863,
7.800000190734863
],
[
0.8999999761581421,
1.899999976158142,
2.9000000953674316,
3.9000000953674316,
4.900000095367432,
5.900000095367432,
6.900000095367432,
7.900000095367432
]
],
"full_dataset_labels": [
0,
1,
2,
0,
1,
2,
0,
1,
2,
0
],
"dtype": "float32",
"torch_version": "2.6.0+cu124",
"format": "Each batch_XXXX.bin file is `[u32 num_tensors=2]` followed by two per-tensor records `[u32 ndim][u32 shape][f32 data]` (little-endian). Tensor 0: features [B, 8]. Tensor 1: labels [B] (label-as-f32). batch_sizes[] records each batch's B.",
"equality_mode": "ORDER"
}