Last active
May 5, 2022 19:06
-
-
Save eddyxu/8b4e9e98ad3ecb3c7744acd3997501fe to your computer and use it in GitHub Desktop.
Load models
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
CREATE MODEL hardhat -- Load from model registry | |
FLAVOR pytorch | |
MODEL_TYPE ssd | |
USING 's3://bucket/to/hardhat.pth'; | |
CREATE MODEL coco_det -- Load pretrained models. | |
FLAVOR pytorch | |
MODEL_TYPE fasterrcnn; |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment