Skip to content
Snippets Groups Projects
main.ipynb 244 KiB
Newer Older
Rishi Sharma's avatar
Rishi Sharma committed
{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "from datasets.Femnist import Femnist\n",
    "from graphs import SmallWorld\n",
    "from collections import defaultdict\n",
    "import os\n",
    "import json\n",
    "import numpy as np\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "a = FEMNIST\n",
    "a"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "b = SmallWorld(6, 2, 2, 1)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "b.adj_list"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "for i in range(12):\n",
    "    print(b.neighbors(i))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "clients = []"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "num_samples = []\n",
    "data = defaultdict(lambda : None)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "datadir = \"./leaf/data/femnist/data/train\"\n",
    "files = os.listdir(datadir)\n",
    "total_users=0\n",
    "users = set()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "files = os.listdir(datadir)[0:1]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "for f in files:\n",
    "    file_path = os.path.join(datadir, f)\n",
    "    print(file_path)\n",
    "    with open(file_path, 'r') as inf:\n",
    "        client_data = json.load(inf)\n",
    "    current_users = len(client_data['users'])\n",
    "    print(\"Current_Users: \", current_users)\n",
    "    total_users += current_users\n",
    "    users.update(client_data['users'])\n",
    "\n",
    "print(\"total_users: \", total_users)\n",
    "print(\"total_users: \", len(users))\n",
    "print(client_data['user_data'].keys())\n",
    "print(np.array(client_data['user_data']['f3408_47']['x']).shape)\n",
    "print(np.array(client_data['user_data']['f3408_47']['y']).shape)\n",
    "print(np.array(client_data['user_data']['f3327_11']['x']).shape)\n",
    "print(np.array(client_data['user_data']['f3327_11']['y']).shape)\n",
    "print(np.unique(np.array(client_data['user_data']['f3327_11']['y'])))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "file = 'run.py'\n",
    "with open(file, 'r') as inf:\n",
    "            print(inf.readline().strip())\n",
    "            print(inf.readlines())"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "def f(l):\n",
    "    l[2] = 'c'\n",
    "\n",
    "a = ['a', 'a', 'a']\n",
    "print(a)\n",
    "f(a)\n",
    "print(a)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "l = ['a', 'b', 'c']\n",
    "print(l[:-1])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "from localconfig import LocalConfig\n",
    "\n",
    "def read_ini(file_path):\n",
    "    config = LocalConfig(file_path)\n",
    "    for section in config:\n",
    "        print(\"Section: \", section)\n",
    "        for key, value in config.items(section):\n",
    "            print((key, value))\n",
    "    print(dict(config.items('DATASET')))\n",
    "    return config\n",
    " \n",
    "config = read_ini(\"config.ini\")\n",
    "for section in config:\n",
    "    print(section)\n",
    "#d = dict(config.sections())"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "def func(a = 1, b = 2, c = 3):\n",
    "    print(a + b + c)\n",
    "\n",
    "l = [3, 5, 7]\n",
    "\n",
    "func(*l)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "from torch import multiprocessing as mp\n",
    "\n",
    "mp.spawn(fn = func, nprocs = 2, args = [], kwargs = {'a': 4, 'b': 5, 'c': 6})"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "l = '[0.4, 0.2, 0.3, 0.1]'\n",
    "type(eval(l))\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "from decentralizepy.datasets.Femnist import Femnist\n",
    "f1 = Femnist(0, 1, 'leaf/data/femnist/data/train')\n",
    "ts = f1.get_trainset(1)\n",
    "for data, target in ts:\n",
    "    print(data)\n",
    "    break"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "from decentralizepy.datasets.Femnist import Femnist\n",
    "from decentralizepy.graphs.SmallWorld import SmallWorld\n",
    "from decentralizepy.mappings.Linear import Linear\n",
    "\n",
    "f = Femnist(2, 'leaf/data/femnist/data/train', sizes=[0.6, 0.4])\n",
    "g = SmallWorld(4, 1, 0.5)\n",
    "l = Linear(2, 2)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "from decentralizepy.node.Node import Node\n",
    "from torch import multiprocessing as mp\n",
    "import logging\n",
    "n1 = Node(0, l, g, f, \"./results\", logging.DEBUG)\n",
    "n2 = Node(1, l, g, f, \"./results\", logging.DEBUG)\n",
    "# mp.spawn(fn = Node, nprocs = 2, args=[l,g,f])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "from testing import f"
   ]
  },
  {
   "cell_type": "code",
Rishi Sharma's avatar
Rishi Sharma committed
   "execution_count": 1,
Rishi Sharma's avatar
Rishi Sharma committed
   "metadata": {},
Rishi Sharma's avatar
Rishi Sharma committed
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "CNN(\n",
      "  (conv1): Conv2d(3, 32, kernel_size=(3, 3), stride=(1, 1), padding=same)\n",
      "  (pool): MaxPool2d(kernel_size=2, stride=2, padding=0, dilation=1, ceil_mode=False)\n",
      "  (conv2): Conv2d(32, 32, kernel_size=(3, 3), stride=(1, 1), padding=same)\n",
      "  (conv3): Conv2d(32, 32, kernel_size=(3, 3), stride=(1, 1), padding=same)\n",
      "  (conv4): Conv2d(32, 32, kernel_size=(3, 3), stride=(1, 1), padding=same)\n",
      "  (fc1): Linear(in_features=800, out_features=2, bias=True)\n",
      ")\n"
     ]
    }
   ],
Rishi Sharma's avatar
Rishi Sharma committed
   "source": [
    "from torch import multiprocessing as mp\n",
    "import torch\n",
Rishi Sharma's avatar
Rishi Sharma committed
    "from decentralizepy.datasets.Celeba import CNN\n",
    "import numpy as np\n",
    "m1 = CNN()\n",
Rishi Sharma's avatar
Rishi Sharma committed
    "o1 = torch.optim.SGD(m1.parameters(), 0.6)\n",
    "print(m1)\n",
    "\n",
    "\n",
Rishi Sharma's avatar
Rishi Sharma committed
    "#mp.spawn(fn = f, nprocs = 2, args=[m1, o1])\n",
Rishi Sharma's avatar
Rishi Sharma committed
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "<generator object Module.parameters at 0x111ec1ba0>"
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "m1.parameters()"
   ]
  },
Rishi Sharma's avatar
Rishi Sharma committed
   "cell_type": "code",
   "execution_count": 25,
Rishi Sharma's avatar
Rishi Sharma committed
   "metadata": {},
Rishi Sharma's avatar
Rishi Sharma committed
327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000
   "outputs": [
    {
     "data": {
      "text/plain": [
       "OrderedDict([('conv1.weight',\n",
       "              tensor([[[[-0.1261,  0.1833, -0.1406],\n",
       "                        [ 0.1324, -0.0685,  0.0938],\n",
       "                        [ 0.0432,  0.1814, -0.0541]],\n",
       "              \n",
       "                       [[-0.1776, -0.1839, -0.0111],\n",
       "                        [ 0.0888,  0.0888, -0.1344],\n",
       "                        [-0.1838,  0.1737,  0.1584]],\n",
       "              \n",
       "                       [[ 0.0417,  0.1064, -0.0156],\n",
       "                        [ 0.0667,  0.0856, -0.1746],\n",
       "                        [ 0.0412,  0.1620,  0.0125]]],\n",
       "              \n",
       "              \n",
       "                      [[[-0.0530, -0.1273, -0.0797],\n",
       "                        [ 0.0422,  0.1135,  0.0475],\n",
       "                        [-0.0244,  0.1691, -0.1383]],\n",
       "              \n",
       "                       [[ 0.0822, -0.1317, -0.1692],\n",
       "                        [ 0.1373,  0.1388,  0.0103],\n",
       "                        [-0.0481,  0.1105,  0.0631]],\n",
       "              \n",
       "                       [[-0.0352,  0.1259, -0.0530],\n",
       "                        [-0.1394, -0.0281,  0.1844],\n",
       "                        [ 0.0082,  0.1187,  0.0211]]],\n",
       "              \n",
       "              \n",
       "                      [[[ 0.0987,  0.0788, -0.1126],\n",
       "                        [ 0.1769,  0.0763, -0.1767],\n",
       "                        [-0.0570,  0.1156,  0.1770]],\n",
       "              \n",
       "                       [[ 0.0643, -0.0024, -0.0625],\n",
       "                        [ 0.0819,  0.0140, -0.1882],\n",
       "                        [ 0.1325, -0.0632, -0.0202]],\n",
       "              \n",
       "                       [[ 0.0053,  0.1042, -0.0058],\n",
       "                        [-0.1082, -0.1753,  0.1762],\n",
       "                        [-0.0501,  0.1166,  0.0561]]],\n",
       "              \n",
       "              \n",
       "                      [[[ 0.0358, -0.0685, -0.1278],\n",
       "                        [ 0.0029, -0.1107,  0.1169],\n",
       "                        [-0.1408,  0.1293,  0.1142]],\n",
       "              \n",
       "                       [[-0.0814,  0.0470,  0.0188],\n",
       "                        [ 0.1538,  0.0137,  0.1128],\n",
       "                        [-0.1597,  0.1432,  0.1370]],\n",
       "              \n",
       "                       [[ 0.1425,  0.1769, -0.0037],\n",
       "                        [-0.1080, -0.0805, -0.0195],\n",
       "                        [-0.1335, -0.1666,  0.1399]]],\n",
       "              \n",
       "              \n",
       "                      [[[ 0.1117,  0.1918, -0.1666],\n",
       "                        [-0.1392,  0.0086,  0.0172],\n",
       "                        [-0.0721, -0.1711,  0.0344]],\n",
       "              \n",
       "                       [[ 0.1820, -0.0537, -0.0974],\n",
       "                        [ 0.0366, -0.0710,  0.1273],\n",
       "                        [ 0.1132, -0.1594,  0.0878]],\n",
       "              \n",
       "                       [[-0.0874, -0.0401,  0.1827],\n",
       "                        [-0.0301,  0.1205, -0.0396],\n",
       "                        [-0.1143, -0.1007,  0.1561]]],\n",
       "              \n",
       "              \n",
       "                      [[[ 0.1522, -0.0012, -0.1785],\n",
       "                        [-0.1833, -0.1828, -0.1643],\n",
       "                        [-0.1765, -0.1757, -0.0608]],\n",
       "              \n",
       "                       [[-0.0684,  0.0521,  0.1137],\n",
       "                        [-0.0028,  0.0616,  0.0758],\n",
       "                        [-0.1736,  0.0667,  0.1229]],\n",
       "              \n",
       "                       [[ 0.1298, -0.1848, -0.1570],\n",
       "                        [-0.1052, -0.1172, -0.1223],\n",
       "                        [-0.1389, -0.0095, -0.0410]]],\n",
       "              \n",
       "              \n",
       "                      [[[ 0.0213, -0.0975,  0.0964],\n",
       "                        [ 0.0535, -0.0775,  0.0790],\n",
       "                        [-0.1796, -0.1468,  0.1036]],\n",
       "              \n",
       "                       [[-0.0403,  0.0646, -0.0932],\n",
       "                        [ 0.1779, -0.1616,  0.0644],\n",
       "                        [-0.0508, -0.1158, -0.0592]],\n",
       "              \n",
       "                       [[-0.1644, -0.1327,  0.0817],\n",
       "                        [ 0.0320, -0.0213, -0.0946],\n",
       "                        [-0.1106,  0.1463, -0.1642]]],\n",
       "              \n",
       "              \n",
       "                      [[[-0.0985, -0.1160, -0.0915],\n",
       "                        [ 0.1857,  0.0806,  0.1761],\n",
       "                        [-0.0817,  0.1095,  0.0896]],\n",
       "              \n",
       "                       [[-0.0660, -0.1680,  0.1833],\n",
       "                        [ 0.0611,  0.0077, -0.0848],\n",
       "                        [-0.1516,  0.1737,  0.0484]],\n",
       "              \n",
       "                       [[ 0.1434, -0.0732, -0.0904],\n",
       "                        [ 0.0962,  0.1783,  0.0192],\n",
       "                        [ 0.0915,  0.0006,  0.0334]]],\n",
       "              \n",
       "              \n",
       "                      [[[-0.0047,  0.1807, -0.1798],\n",
       "                        [-0.0164,  0.1119, -0.0805],\n",
       "                        [ 0.1855, -0.0681, -0.0187]],\n",
       "              \n",
       "                       [[-0.0069,  0.0491, -0.1868],\n",
       "                        [-0.1609, -0.0316,  0.0150],\n",
       "                        [-0.1605,  0.1506, -0.0074]],\n",
       "              \n",
       "                       [[ 0.0851, -0.1732, -0.1777],\n",
       "                        [ 0.0539, -0.0500, -0.1231],\n",
       "                        [ 0.1654,  0.0342, -0.1904]]],\n",
       "              \n",
       "              \n",
       "                      [[[ 0.0476,  0.0284,  0.1212],\n",
       "                        [-0.1603, -0.1924,  0.0144],\n",
       "                        [ 0.0076, -0.0928, -0.1645]],\n",
       "              \n",
       "                       [[ 0.0215,  0.1845, -0.1034],\n",
       "                        [ 0.1574, -0.1577, -0.0438],\n",
       "                        [-0.1360, -0.0601, -0.1693]],\n",
       "              \n",
       "                       [[-0.0720,  0.0619,  0.1405],\n",
       "                        [ 0.0699, -0.1288,  0.0041],\n",
       "                        [-0.0381, -0.1697, -0.1568]]],\n",
       "              \n",
       "              \n",
       "                      [[[-0.1599,  0.1231, -0.1034],\n",
       "                        [-0.0314,  0.0105, -0.1449],\n",
       "                        [-0.0172, -0.0781,  0.0839]],\n",
       "              \n",
       "                       [[-0.0676,  0.1185, -0.1559],\n",
       "                        [-0.1053, -0.1306,  0.1820],\n",
       "                        [ 0.1584, -0.1370,  0.1828]],\n",
       "              \n",
       "                       [[ 0.0658,  0.1412, -0.0537],\n",
       "                        [-0.1230, -0.1411, -0.0011],\n",
       "                        [-0.1318, -0.0458,  0.1838]]],\n",
       "              \n",
       "              \n",
       "                      [[[-0.0268,  0.1747, -0.1037],\n",
       "                        [ 0.0515, -0.0228, -0.1024],\n",
       "                        [-0.1543, -0.0643, -0.0100]],\n",
       "              \n",
       "                       [[-0.1572, -0.1530,  0.0026],\n",
       "                        [ 0.1463, -0.1233,  0.0470],\n",
       "                        [-0.1595, -0.1108, -0.0654]],\n",
       "              \n",
       "                       [[-0.0521, -0.0094,  0.1544],\n",
       "                        [-0.0505, -0.0332,  0.0048],\n",
       "                        [ 0.0735,  0.1350,  0.0690]]],\n",
       "              \n",
       "              \n",
       "                      [[[ 0.0025,  0.0724,  0.0930],\n",
       "                        [-0.1885,  0.0475,  0.1100],\n",
       "                        [-0.1622,  0.0087, -0.0030]],\n",
       "              \n",
       "                       [[ 0.1032, -0.1425, -0.0620],\n",
       "                        [ 0.1515, -0.0736, -0.1888],\n",
       "                        [-0.1246,  0.1424, -0.0491]],\n",
       "              \n",
       "                       [[ 0.1759, -0.1616,  0.1198],\n",
       "                        [-0.1103,  0.1032,  0.1727],\n",
       "                        [-0.0601,  0.1635,  0.0034]]],\n",
       "              \n",
       "              \n",
       "                      [[[ 0.0301,  0.1517,  0.0657],\n",
       "                        [-0.1368, -0.1165,  0.1193],\n",
       "                        [-0.0962,  0.1451,  0.1099]],\n",
       "              \n",
       "                       [[ 0.1646, -0.1860, -0.1187],\n",
       "                        [-0.1367, -0.0911,  0.1337],\n",
       "                        [-0.0926, -0.0524, -0.0672]],\n",
       "              \n",
       "                       [[-0.1509, -0.1231, -0.0855],\n",
       "                        [ 0.1808, -0.0713,  0.0410],\n",
       "                        [-0.0621, -0.0506,  0.1871]]],\n",
       "              \n",
       "              \n",
       "                      [[[ 0.0888, -0.0874, -0.0826],\n",
       "                        [ 0.0416, -0.0961,  0.0603],\n",
       "                        [ 0.1455,  0.0050,  0.0318]],\n",
       "              \n",
       "                       [[-0.1633,  0.0070, -0.1537],\n",
       "                        [-0.0109,  0.1602, -0.0463],\n",
       "                        [-0.0423, -0.0147, -0.1045]],\n",
       "              \n",
       "                       [[ 0.1640, -0.0997, -0.1662],\n",
       "                        [-0.1074,  0.1549, -0.1905],\n",
       "                        [-0.1708,  0.1624,  0.0219]]],\n",
       "              \n",
       "              \n",
       "                      [[[ 0.0824, -0.1376,  0.1086],\n",
       "                        [ 0.0836,  0.0135,  0.0351],\n",
       "                        [-0.1518,  0.0784, -0.1708]],\n",
       "              \n",
       "                       [[-0.1636, -0.1571,  0.1032],\n",
       "                        [-0.1152,  0.0274, -0.1022],\n",
       "                        [-0.0956, -0.1606, -0.1615]],\n",
       "              \n",
       "                       [[ 0.1307,  0.0419,  0.1924],\n",
       "                        [-0.0599, -0.1296, -0.0448],\n",
       "                        [ 0.0363,  0.0377, -0.0460]]],\n",
       "              \n",
       "              \n",
       "                      [[[-0.1685, -0.1277, -0.0465],\n",
       "                        [ 0.0922, -0.1011,  0.0742],\n",
       "                        [ 0.0053, -0.1456,  0.0135]],\n",
       "              \n",
       "                       [[ 0.1341,  0.0131,  0.1281],\n",
       "                        [-0.1020,  0.1069, -0.0631],\n",
       "                        [-0.0439, -0.1189, -0.1822]],\n",
       "              \n",
       "                       [[ 0.1624, -0.1253,  0.0302],\n",
       "                        [ 0.0709,  0.0767,  0.1453],\n",
       "                        [ 0.0203,  0.1603, -0.1720]]],\n",
       "              \n",
       "              \n",
       "                      [[[-0.1550,  0.1513, -0.1003],\n",
       "                        [ 0.0370,  0.0367, -0.0233],\n",
       "                        [ 0.0916, -0.0871,  0.1579]],\n",
       "              \n",
       "                       [[-0.1900,  0.0314,  0.0865],\n",
       "                        [-0.0197,  0.0296, -0.0048],\n",
       "                        [ 0.0846,  0.1543, -0.0770]],\n",
       "              \n",
       "                       [[-0.0016, -0.0978,  0.1826],\n",
       "                        [-0.0477,  0.0689,  0.1079],\n",
       "                        [ 0.0400,  0.0880,  0.1674]]],\n",
       "              \n",
       "              \n",
       "                      [[[ 0.0145, -0.0447, -0.1742],\n",
       "                        [ 0.0394,  0.0127, -0.1172],\n",
       "                        [ 0.1330, -0.1207,  0.0326]],\n",
       "              \n",
       "                       [[-0.0155, -0.1602,  0.0023],\n",
       "                        [ 0.0789,  0.1648,  0.1781],\n",
       "                        [-0.1468, -0.0481, -0.1260]],\n",
       "              \n",
       "                       [[-0.0139,  0.0848, -0.0536],\n",
       "                        [-0.1581,  0.1130,  0.0717],\n",
       "                        [ 0.0275, -0.0006, -0.0049]]],\n",
       "              \n",
       "              \n",
       "                      [[[-0.0199,  0.0032, -0.1246],\n",
       "                        [ 0.0479,  0.1418, -0.1295],\n",
       "                        [-0.1646, -0.1139, -0.1018]],\n",
       "              \n",
       "                       [[ 0.1475,  0.1413, -0.0354],\n",
       "                        [ 0.0612, -0.1652,  0.0801],\n",
       "                        [-0.1306, -0.0165,  0.1733]],\n",
       "              \n",
       "                       [[ 0.1527,  0.0911, -0.1906],\n",
       "                        [-0.1152,  0.1737,  0.0436],\n",
       "                        [-0.0213, -0.0314, -0.0319]]],\n",
       "              \n",
       "              \n",
       "                      [[[-0.0003, -0.0546, -0.1255],\n",
       "                        [ 0.0914, -0.1414,  0.0542],\n",
       "                        [ 0.1139,  0.0132,  0.0815]],\n",
       "              \n",
       "                       [[-0.0042,  0.0541,  0.1456],\n",
       "                        [ 0.0509, -0.0790,  0.0272],\n",
       "                        [ 0.1419,  0.0992, -0.1448]],\n",
       "              \n",
       "                       [[ 0.0496,  0.0013,  0.0838],\n",
       "                        [-0.0662,  0.0315, -0.1168],\n",
       "                        [-0.0069, -0.1503,  0.0729]]],\n",
       "              \n",
       "              \n",
       "                      [[[ 0.1866,  0.1329, -0.0560],\n",
       "                        [ 0.0026,  0.1533,  0.0326],\n",
       "                        [-0.1161, -0.0323,  0.0053]],\n",
       "              \n",
       "                       [[-0.0243, -0.1823, -0.1657],\n",
       "                        [-0.0107, -0.0832,  0.0029],\n",
       "                        [ 0.0981,  0.1241, -0.1788]],\n",
       "              \n",
       "                       [[-0.0400, -0.0577, -0.0757],\n",
       "                        [-0.0584,  0.0176, -0.1019],\n",
       "                        [-0.1828,  0.1589, -0.0312]]],\n",
       "              \n",
       "              \n",
       "                      [[[-0.1083, -0.1236, -0.0904],\n",
       "                        [-0.1575,  0.0157,  0.0552],\n",
       "                        [-0.0839,  0.1704, -0.1457]],\n",
       "              \n",
       "                       [[-0.1648, -0.0270, -0.0489],\n",
       "                        [-0.1122, -0.0288, -0.0073],\n",
       "                        [-0.1443, -0.1712,  0.0100]],\n",
       "              \n",
       "                       [[-0.1142, -0.1552,  0.1568],\n",
       "                        [ 0.0743, -0.1108, -0.0643],\n",
       "                        [-0.0394, -0.1345,  0.0992]]],\n",
       "              \n",
       "              \n",
       "                      [[[-0.1591,  0.0942, -0.1035],\n",
       "                        [-0.0781,  0.0725, -0.0888],\n",
       "                        [ 0.0959,  0.0213,  0.1222]],\n",
       "              \n",
       "                       [[ 0.1202, -0.0217, -0.0955],\n",
       "                        [-0.1748, -0.1133, -0.0704],\n",
       "                        [-0.0670, -0.1401,  0.1553]],\n",
       "              \n",
       "                       [[ 0.0053, -0.0871, -0.0239],\n",
       "                        [ 0.0961, -0.0547,  0.1741],\n",
       "                        [-0.0570,  0.0477,  0.1853]]],\n",
       "              \n",
       "              \n",
       "                      [[[-0.1115, -0.0183, -0.1302],\n",
       "                        [ 0.1435, -0.0238, -0.0048],\n",
       "                        [ 0.1862, -0.1837,  0.1711]],\n",
       "              \n",
       "                       [[ 0.1375, -0.1798,  0.0818],\n",
       "                        [-0.0792,  0.0820,  0.1373],\n",
       "                        [ 0.1849,  0.0672, -0.1822]],\n",
       "              \n",
       "                       [[ 0.1868, -0.0356,  0.0726],\n",
       "                        [-0.1523, -0.1130,  0.1506],\n",
       "                        [-0.1046,  0.0178,  0.0990]]],\n",
       "              \n",
       "              \n",
       "                      [[[ 0.1321, -0.1641,  0.0411],\n",
       "                        [ 0.0526,  0.0393,  0.0918],\n",
       "                        [-0.1345,  0.0750,  0.0859]],\n",
       "              \n",
       "                       [[-0.0985,  0.1466,  0.1349],\n",
       "                        [-0.1461, -0.1742,  0.0941],\n",
       "                        [-0.1502, -0.1813,  0.0864]],\n",
       "              \n",
       "                       [[-0.1039,  0.1179,  0.1499],\n",
       "                        [-0.0366, -0.0120,  0.0951],\n",
       "                        [ 0.0087,  0.1212, -0.0183]]],\n",
       "              \n",
       "              \n",
       "                      [[[-0.1375,  0.0765, -0.0072],\n",
       "                        [-0.0041,  0.0379, -0.0243],\n",
       "                        [-0.1495,  0.1601,  0.1575]],\n",
       "              \n",
       "                       [[-0.0454,  0.1642,  0.0720],\n",
       "                        [-0.0533,  0.0150,  0.0039],\n",
       "                        [ 0.0194,  0.0113, -0.1194]],\n",
       "              \n",
       "                       [[ 0.0527, -0.0886,  0.0359],\n",
       "                        [ 0.1595,  0.0526, -0.0048],\n",
       "                        [-0.1790, -0.0458, -0.0324]]],\n",
       "              \n",
       "              \n",
       "                      [[[-0.1638,  0.0942,  0.0686],\n",
       "                        [-0.1082, -0.0675,  0.1892],\n",
       "                        [-0.1347, -0.1247,  0.0739]],\n",
       "              \n",
       "                       [[ 0.0595,  0.1504, -0.1657],\n",
       "                        [ 0.0733,  0.0529, -0.1599],\n",
       "                        [ 0.0171, -0.1127, -0.0259]],\n",
       "              \n",
       "                       [[-0.0092,  0.0193,  0.1176],\n",
       "                        [-0.1183,  0.0101,  0.1011],\n",
       "                        [ 0.0648, -0.1897,  0.0782]]],\n",
       "              \n",
       "              \n",
       "                      [[[ 0.0098, -0.1161, -0.0802],\n",
       "                        [-0.1821,  0.0221, -0.1754],\n",
       "                        [-0.1218,  0.0525, -0.0480]],\n",
       "              \n",
       "                       [[ 0.0770,  0.0477,  0.1514],\n",
       "                        [ 0.0374, -0.1075, -0.1026],\n",
       "                        [-0.0581, -0.1011,  0.1241]],\n",
       "              \n",
       "                       [[-0.0567, -0.0163,  0.0374],\n",
       "                        [-0.1739, -0.0579,  0.0704],\n",
       "                        [ 0.1817,  0.1561,  0.1677]]],\n",
       "              \n",
       "              \n",
       "                      [[[-0.0569, -0.0763,  0.0044],\n",
       "                        [-0.1133,  0.0813,  0.1477],\n",
       "                        [ 0.0836,  0.0483, -0.1800]],\n",
       "              \n",
       "                       [[ 0.1343, -0.1590,  0.1177],\n",
       "                        [ 0.1071, -0.1647, -0.0646],\n",
       "                        [ 0.1578, -0.1261,  0.0243]],\n",
       "              \n",
       "                       [[-0.0424, -0.0241, -0.0988],\n",
       "                        [ 0.0023,  0.0029, -0.0291],\n",
       "                        [ 0.0415, -0.0557,  0.1427]]],\n",
       "              \n",
       "              \n",
       "                      [[[-0.1028,  0.1054,  0.1658],\n",
       "                        [-0.0357,  0.1579,  0.1237],\n",
       "                        [ 0.0368,  0.0532, -0.1043]],\n",
       "              \n",
       "                       [[-0.0369,  0.0575, -0.1023],\n",
       "                        [ 0.0635,  0.1015,  0.1112],\n",
       "                        [-0.1235,  0.0467,  0.0908]],\n",
       "              \n",
       "                       [[ 0.1380,  0.0633,  0.1087],\n",
       "                        [-0.1360,  0.0422, -0.1524],\n",
       "                        [ 0.0819,  0.0918, -0.1624]]],\n",
       "              \n",
       "              \n",
       "                      [[[ 0.1584, -0.0218, -0.0236],\n",
       "                        [ 0.1878, -0.1289,  0.1343],\n",
       "                        [ 0.0351,  0.1225, -0.1460]],\n",
       "              \n",
       "                       [[ 0.0690, -0.1439,  0.0056],\n",
       "                        [ 0.0272, -0.0058,  0.0125],\n",
       "                        [ 0.0868, -0.0684, -0.0884]],\n",
       "              \n",
       "                       [[ 0.1045,  0.0583, -0.0870],\n",
       "                        [ 0.0600, -0.0732, -0.1695],\n",
       "                        [ 0.0953,  0.0246,  0.1245]]]])),\n",
       "             ('conv1.bias',\n",
       "              tensor([-1.8698e-01, -7.9379e-06, -1.9277e-02,  5.2182e-02,  7.5716e-02,\n",
       "                      -3.3830e-03, -9.6565e-02,  1.0241e-01, -8.2457e-02, -1.6224e-01,\n",
       "                       1.2980e-01, -8.2256e-02, -7.4655e-02, -3.7980e-02,  8.3407e-02,\n",
       "                      -1.4880e-01,  4.8939e-02,  2.7506e-02,  5.8676e-03, -1.5813e-01,\n",
       "                      -6.2464e-04,  1.0359e-02, -1.5525e-01,  7.9100e-02,  1.6850e-02,\n",
       "                      -1.3809e-01, -6.3393e-02, -5.3843e-02, -1.5219e-02, -1.7365e-01,\n",
       "                       1.7249e-01, -1.1165e-01])),\n",
       "             ('conv2.weight',\n",
       "              tensor([[[[ 4.3750e-02,  4.5533e-02, -2.9410e-02],\n",
       "                        [-4.1395e-02,  5.0397e-04, -1.3265e-02],\n",
       "                        [-4.9851e-02, -1.0518e-02,  5.7710e-02]],\n",
       "              \n",
       "                       [[-5.6332e-02, -4.7168e-03, -4.4627e-02],\n",
       "                        [ 5.3513e-03, -4.0824e-02,  1.8281e-02],\n",
       "                        [ 5.0677e-02, -1.5295e-02, -6.1751e-03]],\n",
       "              \n",
       "                       [[-2.4984e-02,  1.2784e-02, -4.7123e-02],\n",
       "                        [-4.3238e-02,  4.7349e-02, -1.5219e-02],\n",
       "                        [-3.6073e-02,  4.1506e-02, -3.5337e-02]],\n",
       "              \n",
       "                       ...,\n",
       "              \n",
       "                       [[ 5.6048e-02,  1.9129e-03, -1.7200e-02],\n",
       "                        [ 5.8869e-02, -5.1520e-02, -5.3205e-02],\n",
       "                        [-1.3903e-02,  5.1790e-02,  2.2585e-02]],\n",
       "              \n",
       "                       [[ 1.1835e-02, -4.9313e-02, -3.1838e-02],\n",
       "                        [ 7.6813e-03,  4.2715e-02, -5.7404e-02],\n",
       "                        [-4.1474e-02, -2.3128e-02, -4.7935e-02]],\n",
       "              \n",
       "                       [[-2.1860e-02, -2.1817e-02, -3.2578e-02],\n",
       "                        [ 3.1317e-02,  3.3435e-02,  3.1837e-02],\n",
       "                        [-2.2399e-03,  3.1600e-02,  4.0183e-02]]],\n",
       "              \n",
       "              \n",
       "                      [[[ 1.9610e-04,  5.3780e-02, -4.5810e-02],\n",
       "                        [ 4.0340e-02,  1.4904e-02, -1.5597e-02],\n",
       "                        [-4.6080e-02,  5.0714e-02, -5.7445e-03]],\n",
       "              \n",
       "                       [[-3.5281e-02,  3.3011e-02,  4.3343e-02],\n",
       "                        [-4.6263e-02, -5.6184e-02,  5.1245e-03],\n",
       "                        [ 3.6015e-02, -3.3152e-02,  4.6629e-03]],\n",
       "              \n",
       "                       [[ 1.7650e-03, -4.2336e-02,  4.3744e-02],\n",
       "                        [ 2.1655e-02,  5.3759e-02,  1.3719e-03],\n",
       "                        [ 4.2005e-02,  5.3998e-02,  1.9009e-02]],\n",
       "              \n",
       "                       ...,\n",
       "              \n",
       "                       [[ 8.9786e-03, -1.8645e-02, -1.3587e-02],\n",
       "                        [-5.4167e-02,  1.2335e-02, -3.0384e-02],\n",
       "                        [-4.8722e-03, -3.7296e-02, -2.6446e-02]],\n",
       "              \n",
       "                       [[ 1.7580e-02,  3.8462e-02, -5.0269e-02],\n",
       "                        [ 2.6601e-03, -1.1462e-02,  4.7459e-02],\n",
       "                        [-2.8888e-02,  3.4436e-02, -4.9943e-02]],\n",
       "              \n",
       "                       [[-5.0206e-02, -5.6025e-02, -3.6346e-02],\n",
       "                        [-2.4407e-02,  5.3721e-02, -5.4920e-02],\n",
       "                        [ 5.1835e-02, -3.2396e-02,  3.2373e-02]]],\n",
       "              \n",
       "              \n",
       "                      [[[-2.7759e-02, -5.4737e-02, -1.1689e-02],\n",
       "                        [ 3.9462e-02,  2.8649e-02,  5.1776e-02],\n",
       "                        [ 2.4253e-02, -2.8318e-02,  2.7402e-02]],\n",
       "              \n",
       "                       [[ 1.3045e-02, -1.0456e-02,  2.0426e-02],\n",
       "                        [ 2.1949e-02,  4.6817e-02, -5.6093e-02],\n",
       "                        [ 2.7145e-02, -5.5441e-02, -2.0719e-02]],\n",
       "              \n",
       "                       [[ 4.4704e-02, -2.4099e-02, -4.7185e-02],\n",
       "                        [-4.3257e-02, -3.3058e-02, -8.6451e-03],\n",
       "                        [-3.7283e-02, -3.4569e-02, -7.1049e-03]],\n",
       "              \n",
       "                       ...,\n",
       "              \n",
       "                       [[-4.1559e-02, -2.9240e-02,  2.7197e-03],\n",
       "                        [ 2.0770e-02,  5.4479e-02, -4.4845e-02],\n",
       "                        [-1.1641e-02, -2.9814e-02, -2.4419e-02]],\n",
       "              \n",
       "                       [[-1.5743e-02,  1.0854e-02,  3.0878e-02],\n",
       "                        [ 2.2739e-02,  3.2999e-02, -1.1902e-02],\n",
       "                        [-3.4837e-02,  1.5305e-02, -8.7552e-03]],\n",
       "              \n",
       "                       [[-2.2882e-02,  9.4639e-03,  5.1878e-03],\n",
       "                        [-2.6344e-02,  2.9063e-02, -1.9337e-02],\n",
       "                        [-3.4314e-02,  1.5313e-02,  4.1524e-02]]],\n",
       "              \n",
       "              \n",
       "                      ...,\n",
       "              \n",
       "              \n",
       "                      [[[ 9.7454e-03, -3.2903e-03,  1.0696e-02],\n",
       "                        [-4.0918e-02,  1.6352e-02,  1.4646e-02],\n",
       "                        [ 1.2516e-02, -2.1804e-02, -2.5489e-02]],\n",
       "              \n",
       "                       [[-1.6083e-02,  2.5374e-02,  3.1458e-02],\n",
       "                        [-3.1497e-02, -1.9513e-02, -2.1223e-02],\n",
       "                        [ 6.6286e-03,  1.6538e-02, -4.8944e-02]],\n",
       "              \n",
       "                       [[ 2.4808e-02, -2.9520e-02, -4.8227e-02],\n",
       "                        [ 1.7325e-03, -4.7443e-02,  2.3087e-03],\n",
       "                        [-1.0008e-02, -2.0313e-02,  2.9944e-02]],\n",
       "              \n",
       "                       ...,\n",
       "              \n",
       "                       [[ 5.5781e-02, -2.0898e-02, -3.1487e-03],\n",
       "                        [-1.6931e-02,  4.5279e-04, -1.5024e-02],\n",
       "                        [-5.5885e-02,  2.7140e-02, -8.5434e-03]],\n",
       "              \n",
       "                       [[ 1.3970e-02, -3.3131e-02,  4.3112e-02],\n",
       "                        [-3.4956e-02, -5.0144e-02, -1.6391e-02],\n",
       "                        [-9.1003e-03, -2.0204e-02, -1.0226e-03]],\n",
       "              \n",
       "                       [[-4.0053e-02, -5.0194e-02,  5.0405e-02],\n",
       "                        [ 5.4107e-02,  4.2185e-02,  3.4359e-02],\n",
       "                        [ 1.6749e-02, -1.4102e-02,  5.0171e-02]]],\n",
       "              \n",
       "              \n",
       "                      [[[ 2.8229e-02,  5.6586e-02, -3.9617e-03],\n",
       "                        [ 2.9538e-02, -1.2507e-02, -2.5516e-02],\n",
       "                        [-1.5193e-02, -2.9232e-02, -2.0701e-02]],\n",
       "              \n",
       "                       [[-5.8773e-02,  3.3015e-02, -9.4146e-03],\n",
       "                        [ 2.8957e-02,  5.8666e-02,  2.8679e-02],\n",
       "                        [ 1.5249e-02, -1.2246e-03,  1.2230e-03]],\n",
       "              \n",
       "                       [[ 2.6050e-02, -4.6042e-02, -3.4895e-03],\n",
       "                        [ 4.9529e-02,  6.6835e-03, -4.1808e-02],\n",
       "                        [-8.6450e-03, -4.8510e-02, -2.4011e-02]],\n",
       "              \n",
       "                       ...,\n",
       "              \n",
       "                       [[-5.1427e-02,  2.4528e-02, -5.4878e-02],\n",
       "                        [-1.8610e-02,  5.4365e-02,  3.5053e-03],\n",
       "                        [-3.9922e-02,  4.2510e-02, -5.7261e-02]],\n",
       "              \n",
       "                       [[ 4.1938e-02, -4.2039e-02, -1.2487e-02],\n",
       "                        [-1.4090e-02, -3.7895e-02,  1.4394e-02],\n",
       "                        [ 2.2555e-02, -2.7264e-02,  5.6102e-02]],\n",
       "              \n",
       "                       [[ 1.5770e-02,  5.4672e-02, -2.4056e-02],\n",
       "                        [ 5.2089e-02, -2.8859e-02, -2.6499e-03],\n",
       "                        [-5.2122e-02, -3.7436e-02,  3.9897e-02]]],\n",
       "              \n",
       "              \n",
       "                      [[[ 2.7888e-02,  2.9241e-02, -1.9488e-02],\n",
       "                        [ 2.8928e-02,  5.3312e-02, -2.9810e-02],\n",
       "                        [-8.5104e-03,  5.7751e-02, -8.1857e-03]],\n",
       "              \n",
       "                       [[ 4.2649e-02,  3.3158e-03,  4.2879e-02],\n",
       "                        [ 7.7893e-03, -3.2879e-02,  2.7630e-02],\n",
       "                        [ 5.4706e-03,  4.8019e-02,  1.2420e-02]],\n",
       "              \n",
       "                       [[-4.2004e-02, -4.2790e-02,  2.4634e-02],\n",
       "                        [-5.4641e-02,  3.4600e-02,  2.9071e-03],\n",
       "                        [ 2.6470e-02,  4.6701e-02,  3.7158e-02]],\n",
       "              \n",
       "                       ...,\n",
       "              \n",
       "                       [[-1.7641e-02, -2.1205e-02, -5.1504e-02],\n",
       "                        [-7.4737e-03,  5.5061e-02, -2.6397e-03],\n",
       "                        [-4.4653e-02, -3.6719e-02,  3.4420e-06]],\n",
       "              \n",
       "                       [[ 1.6525e-02,  1.7280e-02,  5.4554e-03],\n",
       "                        [ 4.0098e-02,  2.7571e-02, -4.4965e-02],\n",
       "                        [ 6.1493e-03, -5.7754e-02,  1.0513e-02]],\n",
       "              \n",
       "                       [[-5.7615e-02,  3.2921e-02, -1.5900e-02],\n",
       "                        [ 2.0081e-02,  5.4590e-02,  1.1296e-02],\n",
       "                        [-4.5015e-02,  1.1341e-03,  2.6447e-02]]]])),\n",
       "             ('conv2.bias',\n",
       "              tensor([-0.0538, -0.0320, -0.0153,  0.0558,  0.0254,  0.0281, -0.0148,  0.0060,\n",
       "                      -0.0283, -0.0062,  0.0437, -0.0064,  0.0341,  0.0233, -0.0201,  0.0391,\n",
       "                       0.0243,  0.0071,  0.0125, -0.0138, -0.0377, -0.0169, -0.0475, -0.0004,\n",
       "                      -0.0105, -0.0502,  0.0241,  0.0090,  0.0069, -0.0315, -0.0192,  0.0204])),\n",
       "             ('conv3.weight',\n",
       "              tensor([[[[-0.0215, -0.0208, -0.0272],\n",
       "                        [-0.0493, -0.0117, -0.0285],\n",
       "                        [ 0.0515, -0.0041,  0.0126]],\n",
       "              \n",
       "                       [[ 0.0299, -0.0301,  0.0552],\n",
       "                        [ 0.0450,  0.0449, -0.0583],\n",
       "                        [-0.0452, -0.0480, -0.0275]],\n",
       "              \n",
       "                       [[-0.0262, -0.0338,  0.0505],\n",
       "                        [ 0.0146, -0.0364, -0.0044],\n",
       "                        [-0.0102, -0.0051,  0.0017]],\n",
       "              \n",
       "                       ...,\n",
       "              \n",
       "                       [[-0.0367, -0.0468, -0.0586],\n",
       "                        [ 0.0126,  0.0037,  0.0191],\n",
       "                        [-0.0153,  0.0048, -0.0160]],\n",
       "              \n",
       "                       [[-0.0050,  0.0364,  0.0582],\n",
       "                        [ 0.0093, -0.0268, -0.0355],\n",
       "                        [-0.0125,  0.0500,  0.0009]],\n",
       "              \n",
       "                       [[ 0.0237, -0.0211, -0.0130],\n",
       "                        [-0.0489,  0.0118,  0.0387],\n",
       "                        [-0.0006,  0.0301,  0.0283]]],\n",
       "              \n",
       "              \n",
       "                      [[[-0.0391, -0.0464, -0.0158],\n",
       "                        [ 0.0201, -0.0054,  0.0422],\n",
       "                        [ 0.0085, -0.0474, -0.0251]],\n",
       "              \n",
       "                       [[-0.0346,  0.0536, -0.0391],\n",
       "                        [ 0.0244, -0.0263, -0.0073],\n",
       "                        [ 0.0076,  0.0160,  0.0044]],\n",
       "              \n",
       "                       [[-0.0128,  0.0146, -0.0381],\n",
       "                        [-0.0277, -0.0142,  0.0226],\n",
       "                        [ 0.0190,  0.0326, -0.0219]],\n",
       "              \n",
       "                       ...,\n",
       "              \n",
       "                       [[-0.0217,  0.0129,  0.0558],\n",
       "                        [ 0.0164, -0.0292, -0.0467],\n",
       "                        [-0.0296,  0.0205, -0.0300]],\n",
       "              \n",
       "                       [[ 0.0254, -0.0151, -0.0583],\n",
       "                        [ 0.0111, -0.0469, -0.0300],\n",
       "                        [-0.0462,  0.0293, -0.0351]],\n",
       "              \n",
       "                       [[ 0.0401, -0.0251,  0.0160],\n",
       "                        [-0.0160, -0.0195, -0.0065],\n",
       "                        [-0.0519,  0.0351,  0.0357]]],\n",
       "              \n",
       "              \n",
       "                      [[[ 0.0544, -0.0209, -0.0454],\n",
       "                        [ 0.0287, -0.0205, -0.0294],\n",
       "                        [-0.0195, -0.0235, -0.0378]],\n",
       "              \n",
       "                       [[-0.0294, -0.0380,  0.0301],\n",
       "                        [ 0.0360,  0.0367,  0.0458],\n",
       "                        [-0.0189, -0.0017,  0.0145]],\n",
       "              \n",
       "                       [[-0.0297,  0.0567,  0.0276],\n",
       "                        [ 0.0298,  0.0383,  0.0227],\n",
       "                        [ 0.0262,  0.0063,  0.0131]],\n",
       "              \n",
       "                       ...,\n",
       "              \n",
       "                       [[ 0.0002,  0.0432, -0.0247],\n",
       "                        [ 0.0068, -0.0298, -0.0484],\n",
       "                        [-0.0361, -0.0014,  0.0444]],\n",
       "              \n",
       "                       [[-0.0184, -0.0201, -0.0163],\n",
       "                        [-0.0466,  0.0255, -0.0244],\n",
       "                        [ 0.0283,  0.0149, -0.0588]],\n",
       "              \n",
       "                       [[ 0.0323,  0.0392, -0.0254],\n",