resnet_mnist_3epoch_prototype.json

{
  "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"
}