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 import learnware
from learnware.market import EasyMarket, BaseUserInfo from learnware.market import EasyMarket, BaseUserInfo
from learnware.reuse import JobSelectorReuser, AveragingReuser 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__)) curr_root = os.path.dirname(os.path.abspath(__file__))


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


user_spec = specification.RKMETableSpecification()
user_spec = RKMETableSpecification()
user_spec.load(os.path.join(unzip_dir, "svm.json")) user_spec.load(os.path.join(unzip_dir, "svm.json"))
user_info = BaseUserInfo(semantic_spec=user_semantic, stat_info={"RKMETableSpecification": user_spec}) 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"]), int(semantic_spec["Output"]["Dimension"]),
): ):
logger.warning( 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 return self.INVALID_LEARNWARE




Loading…
Cancel
Save