File size: 1,267 Bytes
52d060e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 | {
"provenance": "USER_SPECIFIED prototype retrain from configs/resnet_mnist_existing.json with epochs changed to 3 for faster interactive visualization artifact generation.",
"result_status": "PROTOTYPE_ONLY",
"dataset": "MNIST",
"model": "stage9_resnet18_mnist",
"variant": "binary_with_pim_conv2_5_3epoch_prototype",
"epochs": 3,
"batch_size": 50,
"learning_rate": 0.0001,
"seed": 0,
"train_limit": 60000,
"test_limit": 10000,
"device": "auto",
"optimizer": "Adam",
"loss": "CrossEntropyLoss",
"stem_float": true,
"head_float": true,
"clamp_val": 1.0,
"pim": {
"enabled": true,
"conv_indices": [
2,
3,
4,
5
],
"scope_name": "pim_conv_2_5",
"num_rows": 96,
"num_cols": 128,
"num_cycle": 50,
"noise_sigma": 7.0,
"output_sign": false,
"raw_reduction_output": true,
"binary_activation_after_pim": true,
"output_sign_after_reduction": false,
"backward": "STE",
"mode": "sim_cuda"
},
"export": {
"correct_only": true,
"max_samples": 16,
"include_weights": true,
"include_activations": true,
"include_pim_blocks": true,
"output_block_spatial": [
3,
2
],
"output_channel_block": 4
},
"mode": "sim_cuda"
}
|