Skip to content

Commit

Permalink
log params
Browse files Browse the repository at this point in the history
  • Loading branch information
cxnt committed Dec 25, 2023
1 parent 5e7257e commit f92dcc6
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions serve/src/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -140,6 +140,7 @@ def load_on_device(
)
else:
# ------------------------ #
sly.logger.info(deploy_params)
model_source = deploy_params["model_source"] # "pretrained" / "custom"
self.task_type = deploy_params["task_type"]
weights_file_name = deploy_params["weights_name"]
Expand Down

0 comments on commit f92dcc6

Please sign in to comment.