longnew commited on
Commit
52d060e
·
verified ·
1 Parent(s): 1f1baad

Upload resnet_mnist_3epoch_prototype.json with huggingface_hub

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