Browse Source

Merge remote-tracking branch 'origin/main' into feature/hetero

tags/v0.3.2
Peng Tan 2 years ago
parent
commit
969b2e9dfd
2 changed files with 3 additions and 3 deletions
  1. +2
    -2
      examples/workflow_by_code/main.py
  2. +1
    -1
      learnware/market/easy2/checker.py

+ 2
- 2
examples/workflow_by_code/main.py View File

@@ -12,7 +12,7 @@ from shutil import copyfile, rmtree
import learnware
from learnware.market import EasyMarket, BaseUserInfo
from learnware.reuse import JobSelectorReuser, AveragingReuser
from learnware.specification import generate_rkme_spec
from learnware.specification import generate_rkme_spec, RKMETableSpecification

curr_root = os.path.dirname(os.path.abspath(__file__))

@@ -147,7 +147,7 @@ class LearnwareMarketWorkflow:
with zipfile.ZipFile(zip_path, "r") as zip_obj:
zip_obj.extractall(path=unzip_dir)

user_spec = specification.RKMETableSpecification()
user_spec = RKMETableSpecification()
user_spec.load(os.path.join(unzip_dir, "svm.json"))
user_info = BaseUserInfo(semantic_spec=user_semantic, stat_info={"RKMETableSpecification": user_spec})
(


+ 1
- 1
learnware/market/easy2/checker.py View File

@@ -141,7 +141,7 @@ class EasyStatChecker(BaseChecker):
int(semantic_spec["Output"]["Dimension"]),
):
logger.warning(
f"The learnware [{learnware.id}] output dimention mismatch!, where pred_shape={outputs[0].shape}, model_shape={learnware_model.output_shape}, semantic_shape={(int(semantic_spec['Output']['Dimension']), )}"
f"The learnware [{learnware.id}] output dimension mismatch!, where pred_shape={outputs[0].shape}, model_shape={learnware_model.output_shape}, semantic_shape={(int(semantic_spec['Output']['Dimension']), )}"
)
return self.INVALID_LEARNWARE



Loading…
Cancel
Save