{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "e957187d-9efd-41ac-a96f-5c80582bfba5",
   "metadata": {},
   "outputs": [],
   "source": [
    "import subprocess\n",
    "from concurrent.futures import ThreadPoolExecutor, as_completed"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "b4ad0906-3be6-4361-a2eb-975bf10e3b39",
   "metadata": {},
   "outputs": [],
   "source": [
    "task_name = \"read_write\""
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "9857d742-ffcd-4ad8-a16c-f3cf7ed3ad35",
   "metadata": {},
   "outputs": [],
   "source": [
    "# For variable copy\n",
    "if task_name == \"var_copy\":\n",
    "    experiment1_params = {'layer1': ['TF', 'SSM'], 'layer2': ['TF', 'SSM'], 'd': [12, 24], 'window': [5, 10, 15, 20, 30, 50, 100], 'nh': [1], 'sd': [1]}\n",
    "    experiment2_params = {'layer1': ['TF', 'SSM'], 'layer2': ['TF', 'SSM'], 'd': [2, 4, 6, 8, 10, 12, 15, 20, 24], 'window': [20, 100], 'nh': [1], 'sd': [1]}\n",
    "    experiment3_params = {'layer1': ['TF', 'SSM'], 'layer2': ['TF', 'SSM'], 'd': [12, 24], 'window': [20, 100], 'nh': [2,3,4], 'sd': [1]}\n",
    "    experiment4_params = {'layer1': ['TF', 'SSM'], 'layer2': ['TF', 'SSM'], 'd': [12, 24], 'window': [20, 100], 'nh': [1], 'sd': [2,4,8,16]}\n",
    "    experiment5_params = {'layer1': ['TF-nC'], 'layer2': ['TF-nC'], 'd': [2, 4, 6, 8, 10, 12, 15, 20, 24], 'window': [20, 100], 'nh': [1], 'sd': [1]}\n",
    "\n",
    "# For binary recall\n",
    "if task_name in [\"binary_recall_mix\"]:\n",
    "    experiment1_params = {'layer1': ['TF', 'SSM'], 'layer2': ['TF', 'SSM'], 'd': [6], 'window': [5, 10, 15, 20, 30, 50, 100], 'nh': [1], 'sd': [1]}\n",
    "    experiment2_params = {'layer1': ['TF', 'SSM'], 'layer2': ['TF', 'SSM'], 'd': [2, 4, 8, 10, 12, 15, 20], 'window': [100], 'nh': [1], 'sd': [1]}\n",
    "    experiment3_params = {'layer1': ['TF', 'SSM'], 'layer2': ['TF', 'SSM'], 'd': [6], 'window': [100], 'nh': [2,3], 'sd': [1]}\n",
    "    experiment4_params = {'layer1': ['TF', 'SSM'], 'layer2': ['TF', 'SSM'], 'd': [6], 'window': [100], 'nh': [1], 'sd': [2,4,8,16]}\n",
    "    experiment5_params = {'layer1': ['TF-nC'], 'layer2': ['TF-nC'], 'd': [2, 4, 6, 8, 10, 12, 15, 20], 'window': [100], 'nh': [1], 'sd': [1]}\n",
    "\n",
    "# For associative recall\n",
    "if task_name in [\"assoc_recall\", \"assoc_recall_mk\"]:\n",
    "    experiment1_params = {'layer1': ['TF', 'SSM'], 'layer2': ['TF', 'SSM'], 'd': [12], 'window': [5, 10, 15, 20, 50, 100], 'nh': [1], 'sd': [1]}\n",
    "    experiment2_params = {'layer1': ['TF', 'SSM'], 'layer2': ['TF', 'SSM'], 'd': [2, 4, 6, 8, 10, 12], 'window': [20, 100], 'nh': [1], 'sd': [1]}\n",
    "    experiment3_params = {'layer1': ['TF', 'SSM'], 'layer2': ['TF', 'SSM'], 'd': [12], 'window': [20, 100], 'nh': [2,3], 'sd': [1]}\n",
    "    experiment4_params = {'layer1': ['TF', 'SSM'], 'layer2': ['TF', 'SSM'], 'd': [12], 'window': [20, 100], 'nh': [1], 'sd': [2,4,8,16]}\n",
    "    experiment5_params = {'layer1': ['TF-nC'], 'layer2': ['TF-nC'], 'd': [2, 4, 6, 8, 10, 12], 'window': [20, 100], 'nh': [1], 'sd': [1]}\n",
    "\n",
    "# For read write\n",
    "if task_name in [\"read_write\"]:\n",
    "    experiment1_params = {'layer1': ['TF', 'SSM'], 'layer2': ['TF', 'SSM'], 'd': [12], 'window': [5, 10, 15, 20, 50, 100], 'nh': [1], 'sd': [1]}\n",
    "    experiment2_params = {'layer1': ['TF', 'SSM'], 'layer2': ['TF', 'SSM'], 'd': [2, 4, 6, 8, 10, 12], 'window': [20, 100], 'nh': [1], 'sd': [1]}\n",
    "    experiment3_params = {'layer1': ['TF', 'SSM'], 'layer2': ['TF', 'SSM'], 'd': [12], 'window': [20, 100], 'nh': [2,3], 'sd': [1]}\n",
    "    experiment4_params = {'layer1': ['TF', 'SSM'], 'layer2': ['TF', 'SSM'], 'd': [12], 'window': [20, 100], 'nh': [1], 'sd': [2,4,8,16]}\n",
    "    experiment5_params = {'layer1': ['TF-nC'], 'layer2': ['TF-nC'], 'd': [2, 4, 6, 8, 10, 12], 'window': [20, 100], 'nh': [1], 'sd': [1]}"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "501791fd-5cd8-4086-86bf-655500db978c",
   "metadata": {},
   "outputs": [],
   "source": [
    "# List of commands to run\n",
    "commands = []\n",
    "\n",
    "def make_command(layer1, layer2, embed_dim, window, num_heads, state_dim, run_number=0):\n",
    "    return f\"python3 train.py --run_number {run_number} --task_name {task_name} --layer1 {layer1} --layer2 {layer2} --embed_dim {embed_dim} --window {window} --num_heads {num_heads} --state_dim {state_dim} --save True\"\n",
    "\n",
    "for run_number in range(11):\n",
    "    for experiment in [experiment1_params, experiment2_params, experiment3_params, experiment4_params, experiment5_params]:\n",
    "        for layer1 in experiment[\"layer1\"]:\n",
    "            for layer2 in experiment[\"layer2\"]:\n",
    "                for embed_dim in experiment[\"d\"]:\n",
    "                    for window in experiment[\"window\"]:\n",
    "                        for num_heads in experiment[\"nh\"]:\n",
    "                            for state_dim in experiment[\"sd\"]:\n",
    "                                commands.append(make_command(layer1, layer2, embed_dim, window, num_heads, state_dim, run_number))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "id": "b9e6a6e5-bfbd-44b8-9c76-591e8cd47ec4",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "['python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 0 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 1 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 2 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 3 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 4 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 5 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 6 --task_name read_write --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True',\n",
       " 'python3 train.py --run_number 7 --task_name read_write --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True',\n",
       " ...]"
      ]
     },
     "execution_count": 7,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "\n",
    "# commands = ['python3 train.py --run_number 2 --task_name var_copy --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True']\n",
    "commands"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "30f4f897-7954-4843-a4a3-c1abe3a66ea3",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n",
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 0 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n",
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n",
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 1 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n",
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n",
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 2 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n",
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n",
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 3 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n",
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n",
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 4 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n",
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 5 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n",
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n",
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 6 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n",
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n",
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 7 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n",
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n",
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 8 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n",
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n",
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n",
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 9 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 5 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 10 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 15 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 50 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n",
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 2 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 3 --state_dim 1 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n",
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 2 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 4 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 6 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 8 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 10 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 20 --num_heads 1 --state_dim 1 --save True] exited with 2\n",
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 TF-nC --layer2 TF-nC --embed_dim 12 --window 100 --num_heads 1 --state_dim 1 --save True] exited with 2\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 TF --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 20 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 4 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 2 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 8 --save True] exited with 0\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[python3 train.py --run_number 10 --task_name assoc_recall_mk --layer1 SSM --layer2 SSM --embed_dim 12 --window 100 --num_heads 1 --state_dim 16 --save True] exited with 0\n"
     ]
    }
   ],
   "source": [
    "def run_command(cmd):\n",
    "    \"\"\"Run a single shell command and return (cmd, returncode, stdout, stderr).\"\"\"\n",
    "    result = subprocess.run(cmd, shell=True, capture_output=True, text=True)\n",
    "    return cmd, result.returncode, result.stdout, result.stderr\n",
    "\n",
    "max_workers = 20\n",
    "\n",
    "with ThreadPoolExecutor(max_workers=max_workers) as executor:\n",
    "    futures = {executor.submit(run_command, cmd): cmd for cmd in commands}\n",
    "\n",
    "    for future in as_completed(futures):\n",
    "        cmd, returncode, stdout, stderr = future.result()\n",
    "        print(f\"[{cmd}] exited with {returncode}\")\n",
    "        if returncode == 1:\n",
    "            executor.shutdown()\n",
    "            print(stdout)\n",
    "            print(stderr)\n",
    "            assert False"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "d4bc2226-523f-42de-9fc1-d46f96103d4d",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "hybrid",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.12.11"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}