Add matplotlib and tensorboard requirements

2021-05-06 12:38:57 -07:00 · 2021-05-06 12:38:57 -07:00 · 413221af7b
commit 413221af7b
parent 2717269a8d
1 changed files with 2 additions and 3 deletions
--- a/setup.py
+++ b/setup.py
@ -15,8 +15,7 @@ setup(
        'torch>=1.2',
        'numpy',
        'scipy',
        'matplotlib',
        'tensorboard',
    ],
    extras_require={
        "visualization":  ["tensorboard"],
    },
 )