File tree Expand file tree Collapse file tree 2 files changed +2
-9
lines changed Expand file tree Collapse file tree 2 files changed +2
-9
lines changed Original file line number Diff line number Diff line change @@ -8,15 +8,16 @@ It demonstrates how a neural network with convolutional and fully connected laye
8
8
numpy
9
9
scipy
10
10
scikit-tensor-py3
11
+ absl-py
11
12
flopco-pytorch
13
+ tensorly-musco
12
14
pytorch
13
15
```
14
16
15
17
## Installation
16
18
```
17
19
pip install musco-pytorch
18
20
```
19
- If something is going wrong, try ``` pip install git+https://github.com/musco-ai/musco-pytorch ```
20
21
21
22
## Quick Start
22
23
``` python
Original file line number Diff line number Diff line change 2
2
from setuptools import setup , find_packages
3
3
from setuptools .command .install import install
4
4
5
-
6
- class InstallLocalPackage (install ):
7
- def run (self ):
8
- install .run (self )
9
- subprocess .call ("cd dependencies/tensorly && python setup.py install --force && cd ../.." , shell = True )
10
-
11
-
12
5
try :
13
6
from pip ._internal .req import parse_requirements
14
7
except ImportError :
@@ -30,6 +23,5 @@ def load_requirements(file_name):
30
23
download_url = "https://github.com/musco-ai/musco-pytorch/archive/v1.0.2.tar.gz" ,
31
24
license = "Apache-2.0" ,
32
25
packages = find_packages (),
33
- cmdclass = {"install" : InstallLocalPackage },
34
26
install_requires = load_requirements ("requirements.txt" )
35
27
)
You can’t perform that action at this time.
0 commit comments