Change tensorboard to optional dependency
This commit is contained in:
parent
6c030f1eb1
commit
6252629631
1 changed files with 3 additions and 1 deletions
4
setup.py
4
setup.py
|
@ -13,8 +13,10 @@ setup(
|
||||||
'torch',
|
'torch',
|
||||||
'numpy',
|
'numpy',
|
||||||
'scipy',
|
'scipy',
|
||||||
'tensorboard',
|
|
||||||
],
|
],
|
||||||
|
extras_require={
|
||||||
|
"visualization": ["tensorboard"],
|
||||||
|
}
|
||||||
scripts=[
|
scripts=[
|
||||||
'scripts/m2m.py',
|
'scripts/m2m.py',
|
||||||
]
|
]
|
||||||
|
|
Loading…
Add table
Reference in a new issue