KordingLab
diff --git a/‎analysis/Demo Analysis.ipynb
+4-4 b/‎analysis/Demo Analysis.ipynb
+4-4
diff --git a/‎analysis/Untitled.ipynb
-399 b/‎analysis/Untitled.ipynb
-399
diff --git a/‎analysis/Untitled1.ipynb
-68 b/‎analysis/Untitled1.ipynb
-68
diff --git a/‎analysis/graphs/inhibition.svg
-21 b/‎analysis/graphs/inhibition.svg
-21
diff --git a/‎coco_attention_model3_log.txt
-3 b/‎coco_attention_model3_log.txt
-3
diff --git a/‎saved/plots/train/coco_attention_model3_masked_0.png
-94.6 KB b/‎saved/plots/train/coco_attention_model3_masked_0.png
-94.6 KB
diff --git a/‎saved/plots/train/coco_attention_model3_masked_1.png
-80.3 KB b/‎saved/plots/train/coco_attention_model3_masked_1.png
-80.3 KB
diff --git a/‎saved/plots/train/coco_attention_model3_orig.png.png
-87.6 KB b/‎saved/plots/train/coco_attention_model3_orig.png.png
-87.6 KB
diff --git a/‎saved/plots/val/coco_attention_model3_masked_0.png
-105 KB b/‎saved/plots/val/coco_attention_model3_masked_0.png
-105 KB
diff --git a/‎saved/plots/val/coco_attention_model3_masked_1.png
-89.4 KB b/‎saved/plots/val/coco_attention_model3_masked_1.png
-89.4 KB
diff --git a/‎saved/plots/val/coco_attention_model3_orig.png.png
-97.7 KB b/‎saved/plots/val/coco_attention_model3_orig.png.png
-97.7 KB
diff --git a/‎train_coco.py
+2-2 b/‎train_coco.py
+2-2
@@ -66,8 +66,8 @@
    "source": [
     "modelpath = \"../saved/models/paper_coco_model.pt\" #path to the saved model\n",
     "modeltype = \"coco\" #type of model: 'mnist' or 'coco'\n",
-    "cocoroot = '../../../../data/jordanlei/coco/images/val2017' #path to the coco val dataset\n",
-    "annpath = '../../../../data/jordanlei/coco/annotations/instances_val2017.json' #path to the coco val annotations\n",
+    "cocoroot = '../../../../data/coco/images/val2017' #path to the coco val dataset\n",
+    "annpath = '../../../../data/coco/annotations/instances_val2017.json' #path to the coco val annotations\n",
     "metadatapath = '../data/metadata/cocometadata_test.p' #path to metadata file (will create one if none exists)\n",
     "\n",
     "n = 2 #number of objects\n",
@@ -499,9 +499,9 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python (jordan_env)",
+   "display_name": "Python (env)",
    "language": "python",
-   "name": "jordan_env"
+   "name": "env"
   },
   "language_info": {
    "codemirror_mode": {
 
@@ -4,8 +4,8 @@
 
 parser = argparse.ArgumentParser()
 parser.add_argument('--device', type= int, default = 3)
-parser.add_argument('--trainpath', type= str, default = '../../../data/jordanlei/coco/images/train2017')
-parser.add_argument('--annpath', type= str, default = '../../../data/jordanlei/coco/annotations/instances_train2017.json')
+parser.add_argument('--trainpath', type= str, default = '../../../data/coco/images/train2017')
+parser.add_argument('--annpath', type= str, default = '../../../data/coco/annotations/instances_train2017.json')
 parser.add_argument('--metadatapath', type= str, default = 'data/metadata/cocometadata_train.p')
 parser.add_argument('--strength', type= float, default = 0.3)
 parser.add_argument('--out', type = str, default='attn')