This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
AI流水线
Repositories
Datasets
Forum
实训
竞赛
大数据
AI开发
Register
Sign In
beimingwu
/
learnware
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
1
Wiki
evaluate
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
HPC
Browse Source
[FIX] fix import bugs, and update setup
tags/v0.3.2
bxdd
2 years ago
parent
b6a61df769
commit
ac2c4bb4c5
2 changed files
with
4 additions
and
3 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+1
-1
learnware/specification/system/__init__.py
+3
-2
setup.py
+ 1
- 1
learnware/specification/system/__init__.py
View File
@@ -8,4 +8,4 @@ if not is_torch_available(verbose=False):
HeteroMapTableSpecification = None
logger.warning("HeteroMapTableSpecification is skipped because torch is not installed!")
else:
from .
rkm
e import HeteroMapTableSpecification
from .
hetero_tabl
e import HeteroMapTableSpecification
+ 3
- 2
setup.py
View File
@@ -107,8 +107,9 @@ if __name__ == "__main__":
"pre-commit",
],
"full": [
"torch>=2.0.0",
"torchvision>=0.15.1",
# The default full requirements for learnware package
"torch==2.1.0",
"torchvision==0.16.0",
"torch-optimizer>=0.3.0",
"lightgbm>=3.3.0",
"sentence_transformers>=2.2.2",
Write
Preview
Loading…
Cancel
Save