mirror of
https://gitee.com/mymagicpower/AIAS.git
synced 2024-11-29 18:58:16 +08:00
no message
This commit is contained in:
parent
48a608bb87
commit
392354dd46
@ -58,7 +58,7 @@ public final class Inceptionv3ActionRecognition {
|
||||
// .optModelUrls("/Users/calvin/Documents/build/mxnet_models/inceptionv3_ucf101/")
|
||||
.optTranslator(translator)
|
||||
.optProgress(new ProgressBar())
|
||||
.optDevice(Device.cpu())
|
||||
//.optDevice(Device.cpu())
|
||||
.build();
|
||||
|
||||
try (ZooModel<Image, Classifications> inception = ModelZoo.loadModel(criteria)) {
|
||||
|
@ -58,7 +58,7 @@ public final class Vgg16ActionRecognition {
|
||||
// .optModelUrls("/Users/calvin/Documents/build/mxnet_models/vgg16_ucf101/")
|
||||
.optTranslator(translator)
|
||||
.optProgress(new ProgressBar())
|
||||
.optDevice(Device.cpu())
|
||||
//.optDevice(Device.cpu())
|
||||
.build();
|
||||
|
||||
try (ZooModel<Image, Classifications> inception = ModelZoo.loadModel(criteria)) {
|
||||
|
@ -30,7 +30,7 @@
|
||||
<orderEntry type="library" name="Maven: ai.djl:model-zoo:0.14.0" level="project" />
|
||||
<orderEntry type="library" name="Maven: ai.djl.pytorch:pytorch-engine:0.14.0" level="project" />
|
||||
<orderEntry type="library" name="Maven: ai.djl.pytorch:pytorch-model-zoo:0.14.0" level="project" />
|
||||
<orderEntry type="library" name="Maven: ai.djl.pytorch:pytorch-native-auto:1.9.0" level="project" />
|
||||
<orderEntry type="library" name="Maven: ai.djl.pytorch:pytorch-native-auto:1.9.1" level="project" />
|
||||
<orderEntry type="library" name="Maven: ai.djl.paddlepaddle:paddlepaddle-model-zoo:0.14.0" level="project" />
|
||||
<orderEntry type="library" name="Maven: ai.djl.paddlepaddle:paddlepaddle-engine:0.14.0" level="project" />
|
||||
<orderEntry type="library" name="Maven: ai.djl.paddlepaddle:paddlepaddle-native-auto:2.0.2" level="project" />
|
||||
|
@ -97,7 +97,7 @@
|
||||
<dependency>
|
||||
<groupId>ai.djl.pytorch</groupId>
|
||||
<artifactId>pytorch-native-auto</artifactId>
|
||||
<version>1.9.0</version>
|
||||
<version>1.9.1</version>
|
||||
</dependency>
|
||||
<!-- PaddlePaddle -->
|
||||
<dependency>
|
||||
|
@ -71,7 +71,7 @@ public final class AnimalsClassification {
|
||||
.optModelName("inference")
|
||||
.optTranslator(new AnimalTranslator())
|
||||
.optProgress(new ProgressBar())
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.build();
|
||||
|
||||
try (ZooModel rotateModel = ModelZoo.loadModel(criteria)) {
|
||||
|
@ -72,7 +72,7 @@ public final class LightAnimalsClassification {
|
||||
.optModelName("inference")
|
||||
.optTranslator(new AnimalTranslator())
|
||||
.optProgress(new ProgressBar())
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.build();
|
||||
|
||||
try (ZooModel rotateModel = ModelZoo.loadModel(criteria)) {
|
||||
|
@ -98,7 +98,7 @@
|
||||
<dependency>
|
||||
<groupId>ai.djl.pytorch</groupId>
|
||||
<artifactId>pytorch-native-auto</artifactId>
|
||||
<version>1.9.0</version>
|
||||
<version>1.9.1</version>
|
||||
</dependency>
|
||||
<!-- PaddlePaddle -->
|
||||
<dependency>
|
||||
|
@ -68,7 +68,7 @@ public final class DishesClassification {
|
||||
.optModelName("inference")
|
||||
.optTranslator(new DishTranslator())
|
||||
.optProgress(new ProgressBar())
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.build();
|
||||
|
||||
try (ZooModel model = ModelZoo.loadModel(criteria)) {
|
||||
|
@ -72,7 +72,7 @@ public final class LightDishesClassification {
|
||||
.optModelName("inference")
|
||||
.optTranslator(new DishTranslator())
|
||||
.optProgress(new ProgressBar())
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.build();
|
||||
|
||||
try (ZooModel model = ModelZoo.loadModel(criteria)) {
|
||||
|
@ -97,7 +97,7 @@
|
||||
<dependency>
|
||||
<groupId>ai.djl.pytorch</groupId>
|
||||
<artifactId>pytorch-native-auto</artifactId>
|
||||
<version>1.9.0</version>
|
||||
<version>1.9.1</version>
|
||||
</dependency>
|
||||
<!-- PaddlePaddle -->
|
||||
<dependency>
|
||||
|
@ -60,7 +60,7 @@ public final class PersonDetection {
|
||||
// .optModelUrls("/Users/calvin/Documents/build/mxnet_models/ssd_512_resnet50_v1_voc/")
|
||||
.optTranslator(translator)
|
||||
.optProgress(new ProgressBar())
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.build();
|
||||
|
||||
try (ZooModel<Image, DetectedObjects> ssd = ModelZoo.loadModel(criteria)) {
|
||||
|
@ -56,7 +56,7 @@ public final class PoseResnet18Estimation {
|
||||
// .optModelUrls("/Users/calvin/Documents/build/mxnet_models/simple_pose_resnet18_v1b/")
|
||||
.optTranslator(translator)
|
||||
.optProgress(new ProgressBar())
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.build();
|
||||
|
||||
try (ZooModel<Image, Joints> pose = ModelZoo.loadModel(criteria);
|
||||
|
@ -56,7 +56,7 @@ public final class PoseResnet50Estimation {
|
||||
// .optModelUrls("/Users/calvin/Documents/build/mxnet_models/simple_pose_resnet50_v1b/")
|
||||
.optTranslator(translator)
|
||||
.optProgress(new ProgressBar())
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.build();
|
||||
|
||||
try (ZooModel<Image, Joints> pose = ModelZoo.loadModel(criteria);
|
||||
|
@ -34,7 +34,7 @@ public final class CrowdDetect {
|
||||
// .optModelName("inference")
|
||||
.optTranslator(new CrowdTranslator())
|
||||
.optProgress(new ProgressBar())
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.build();
|
||||
|
||||
return criteria;
|
||||
|
@ -38,7 +38,7 @@ public final class FaceDetection {
|
||||
// .optModelUrls("/Users/calvin/model/face_mask/pyramidbox_lite/")
|
||||
// .optModelName("inference")
|
||||
.optProgress(new ProgressBar())
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.optTranslator(new FaceTranslator(shrink, threshold))
|
||||
.build();
|
||||
|
||||
|
@ -29,7 +29,7 @@ public final class FaceLandmark {
|
||||
// .optModelUrls("/Users/calvin/model/face_landmark/face_landmark/")
|
||||
.optModelName("inference")
|
||||
.optProgress(new ProgressBar())
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.optTranslator(new FaceLandmarkTranslator())
|
||||
.build();
|
||||
|
||||
|
@ -38,7 +38,7 @@ public final class FaceDetection {
|
||||
// .optModelUrls("/Users/calvin/model/face_mask/pyramidbox_lite/")
|
||||
// .optModelName("inference")
|
||||
.optProgress(new ProgressBar())
|
||||
.optDevice(Device.cpu())
|
||||
//.optDevice(Device.cpu())
|
||||
.optTranslator(new FaceTranslator(shrink, threshold))
|
||||
.build();
|
||||
|
||||
|
@ -38,7 +38,7 @@ public final class FaceDetection {
|
||||
// .optModelUrls("/Users/calvin/model/face_mask/pyramidbox_lite/")
|
||||
// .optModelName("inference")
|
||||
.optProgress(new ProgressBar())
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.optTranslator(new FaceTranslator(shrink, threshold))
|
||||
.build();
|
||||
|
||||
|
@ -66,7 +66,7 @@ public final class FaceMaskDetect {
|
||||
.optModelUrls(
|
||||
"https://aias-home.oss-cn-beijing.aliyuncs.com/models/face_mask/face_mask.zip")
|
||||
.optProgress(new ProgressBar())
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.build();
|
||||
|
||||
return criteria;
|
||||
|
@ -42,7 +42,7 @@ public final class PedestrianDetect {
|
||||
.optModelName("inference")
|
||||
.optTranslator(new PedestrianTranslator())
|
||||
.optProgress(new ProgressBar())
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.build();
|
||||
|
||||
return criteria;
|
||||
|
@ -43,7 +43,7 @@ public final class LargeSafetyHelmetDetect {
|
||||
.optTranslator(translator)
|
||||
.optProgress(new ProgressBar())
|
||||
.optEngine("MXNet")
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.build();
|
||||
|
||||
return criteria;
|
||||
|
@ -43,7 +43,7 @@ public final class MediumSafetyHelmetDetect {
|
||||
.optTranslator(translator)
|
||||
.optProgress(new ProgressBar())
|
||||
.optEngine("MXNet")
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.build();
|
||||
|
||||
return criteria;
|
||||
|
@ -43,7 +43,7 @@ public final class SmallSafetyHelmetDetect {
|
||||
.optTranslator(translator)
|
||||
.optProgress(new ProgressBar())
|
||||
.optEngine("MXNet")
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.build();
|
||||
|
||||
return criteria;
|
||||
|
@ -72,7 +72,7 @@ public final class TrafficDetection {
|
||||
.optModelName("inference")
|
||||
.optTranslator(new TrafficTranslator())
|
||||
.optProgress(new ProgressBar())
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.build();
|
||||
|
||||
try (ZooModel model = ModelZoo.loadModel(criteria)) {
|
||||
|
@ -46,7 +46,7 @@ public final class VehicleDetect {
|
||||
.optModelName("inference")
|
||||
.optTranslator(new VehicleTranslator())
|
||||
.optProgress(new ProgressBar())
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.build();
|
||||
|
||||
return criteria;
|
||||
|
@ -24,7 +24,7 @@ public final class LayoutDetection {
|
||||
"https://aias-home.oss-cn-beijing.aliyuncs.com/models/ocr_models/ppyolov2_r50vd_dcn_365e_publaynet_infer.zip")
|
||||
// .optModelUrls(
|
||||
// "/Users/calvin/.paddledet/inference_model/ppyolov2_r50vd_dcn_365e_publaynet/ppyolov2_r50vd_dcn_365e_publaynet_infer")
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.optTranslator(new LayoutDetectionTranslator())
|
||||
.optProgress(new ProgressBar())
|
||||
.build();
|
||||
|
@ -78,7 +78,7 @@ public final class LightOcrDetection {
|
||||
"https://aias-home.oss-cn-beijing.aliyuncs.com/models/ocr_models/ch_PP-OCRv2_det_infer.zip")
|
||||
// .optModelUrls(
|
||||
// "/Users/calvin/Documents/build/paddle_models/ppocr/ch_PP-OCRv2_det_infer")
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.optTranslator(new PpWordDetectionTranslator(new ConcurrentHashMap<String, String>()))
|
||||
.optProgress(new ProgressBar())
|
||||
.build();
|
||||
|
@ -62,7 +62,7 @@ public final class LightOcrRecognition {
|
||||
.setTypes(Image.class, DetectedObjects.class)
|
||||
.optModelUrls(
|
||||
"https://aias-home.oss-cn-beijing.aliyuncs.com/models/ocr_models/ch_PP-OCRv2_det_infer.zip")
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.optProgress(new ProgressBar())
|
||||
.optTranslator(new PpWordDetectionTranslator(new ConcurrentHashMap<String, String>()))
|
||||
.build();
|
||||
@ -77,7 +77,7 @@ public final class LightOcrRecognition {
|
||||
.setTypes(Image.class, String.class)
|
||||
.optModelUrls(
|
||||
"https://aias-home.oss-cn-beijing.aliyuncs.com/models/ocr_models/ch_PP-OCRv2_rec_infer.zip")
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.optProgress(new ProgressBar())
|
||||
.optTranslator(new PpWordRecognitionTranslator())
|
||||
.build();
|
||||
|
@ -78,7 +78,7 @@ public final class MobileOcrDetection {
|
||||
"https://aias-home.oss-cn-beijing.aliyuncs.com/models/ocr_models/ch_ppocr_mobile_v2.0_det_infer.zip")
|
||||
// .optModelUrls(
|
||||
// "/Users/calvin/Documents/build/paddle_models/ppocr/ch_PP-OCRv2_det_infer")
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.optTranslator(new PpWordDetectionTranslator(new ConcurrentHashMap<String, String>()))
|
||||
.optProgress(new ProgressBar())
|
||||
.build();
|
||||
|
@ -62,7 +62,7 @@ public final class MobileOcrRecognition {
|
||||
"https://aias-home.oss-cn-beijing.aliyuncs.com/models/ocr_models/ch_ppocr_mobile_v2.0_det_infer.zip")
|
||||
// .optModelUrls(
|
||||
// "/Users/calvin/Documents/build/paddle_models/ppocr/ch_PP-OCRv2_det_infer")
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.optTranslator(new PpWordDetectionTranslator(new ConcurrentHashMap<String, String>()))
|
||||
.optProgress(new ProgressBar())
|
||||
.build();
|
||||
@ -77,7 +77,7 @@ public final class MobileOcrRecognition {
|
||||
.setTypes(Image.class, String.class)
|
||||
.optModelUrls(
|
||||
"https://aias-home.oss-cn-beijing.aliyuncs.com/models/ocr_models/ch_ppocr_mobile_v2.0_rec_infer.zip")
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.optProgress(new ProgressBar())
|
||||
.optTranslator(new PpWordRecognitionTranslator())
|
||||
.build();
|
||||
|
@ -77,7 +77,7 @@ public final class ServerOcrDetection {
|
||||
"https://aias-home.oss-cn-beijing.aliyuncs.com/models/ocr_models/ch_ppocr_server_v2.0_det_infer.zip")
|
||||
// .optModelUrls(
|
||||
// "/Users/calvin/Documents/build/paddle_models/ppocr/ch_ppocr_server_v2.0_det_infer")
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.optTranslator(new PpWordDetectionTranslator(new ConcurrentHashMap<String, String>()))
|
||||
.optProgress(new ProgressBar())
|
||||
.build();
|
||||
@ -94,7 +94,7 @@ public final class ServerOcrDetection {
|
||||
"https://aias-home.oss-cn-beijing.aliyuncs.com/models/ocr_models/ch_ppocr_mobile_v2.0_cls_infer.zip")
|
||||
// .optModelUrls(
|
||||
// "/Users/calvin/Documents/build/paddle_models/ppocr/ch_ppocr_mobile_v2.0_cls_infer")
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.optTranslator(new PpWordRotateTranslator())
|
||||
.optProgress(new ProgressBar())
|
||||
.build();
|
||||
|
@ -59,7 +59,7 @@ public final class ServerOcrRecognition {
|
||||
.setTypes(Image.class, DetectedObjects.class)
|
||||
.optModelUrls(
|
||||
"https://aias-home.oss-cn-beijing.aliyuncs.com/models/ocr_models/ch_ppocr_server_v2.0_det_infer.zip")
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.optProgress(new ProgressBar())
|
||||
.optTranslator(new PpWordDetectionTranslator(new ConcurrentHashMap<String, String>()))
|
||||
.build();
|
||||
@ -74,7 +74,7 @@ public final class ServerOcrRecognition {
|
||||
.setTypes(Image.class, String.class)
|
||||
.optModelUrls(
|
||||
"https://aias-home.oss-cn-beijing.aliyuncs.com/models/ocr_models/ch_ppocr_server_v2.0_rec_infer.zip")
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.optProgress(new ProgressBar())
|
||||
.optTranslator(new PpWordRecognitionTranslator())
|
||||
.build();
|
||||
|
@ -32,7 +32,7 @@ public final class TableDetection {
|
||||
"https://aias-home.oss-cn-beijing.aliyuncs.com/models/ocr_models/en_table.zip")
|
||||
// .optModelUrls(
|
||||
// "/Users/calvin/Documents/build/paddle_models/ppocr/en_ppocr_mobile_v2.0_table_structure_infer")
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.optOption("removePass", "repeated_fc_relu_fuse_pass")
|
||||
.optTranslator(new TableStructTranslator(new ConcurrentHashMap<String, String>()))
|
||||
.optProgress(new ProgressBar())
|
||||
|
@ -23,7 +23,7 @@ public final class Lac {
|
||||
.optTranslator(new LacTranslator())
|
||||
.optProgress(new ProgressBar())
|
||||
.optEngine("PaddlePaddle") // Use PaddlePaddle engine
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.build();
|
||||
|
||||
return criteria;
|
||||
|
@ -29,7 +29,7 @@ public final class Translation {
|
||||
.optTranslator(new TranslationTranslator())
|
||||
.optEngine("PaddlePaddle") // Use PyTorch engine
|
||||
.optModelName("inference")
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.optProgress(new ProgressBar())
|
||||
.build();
|
||||
|
||||
|
@ -23,7 +23,7 @@ public final class Lac {
|
||||
.optTranslator(new LacTranslator())
|
||||
.optProgress(new ProgressBar())
|
||||
.optEngine("PaddlePaddle") // Use PaddlePaddle engine
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.build();
|
||||
|
||||
return criteria;
|
||||
|
@ -24,7 +24,7 @@ public final class SimnetBow {
|
||||
// .optModelUrls("/Users/calvin/model/simnet_bow/")
|
||||
.optTranslator(new SimnetBowTranslator())
|
||||
.optEngine("PaddlePaddle") // Use PyTorch engine
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.optProgress(new ProgressBar())
|
||||
.build();
|
||||
|
||||
|
@ -23,7 +23,7 @@ public final class Lac {
|
||||
.optTranslator(new LacTranslator())
|
||||
.optProgress(new ProgressBar())
|
||||
.optEngine("PaddlePaddle") // Use PaddlePaddle engine
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.build();
|
||||
|
||||
return criteria;
|
||||
|
@ -29,7 +29,7 @@ public final class Translation {
|
||||
.optTranslator(new TranslationTranslator())
|
||||
.optEngine("PaddlePaddle") // Use PyTorch engine
|
||||
.optModelName("inference")
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.optProgress(new ProgressBar())
|
||||
.build();
|
||||
|
||||
|
@ -23,7 +23,7 @@ public final class Lac {
|
||||
.optTranslator(new LacTranslator())
|
||||
.optProgress(new ProgressBar())
|
||||
.optEngine("PaddlePaddle") // Use PaddlePaddle engine
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.build();
|
||||
|
||||
return criteria;
|
||||
|
@ -29,7 +29,7 @@ public final class Translation {
|
||||
.optTranslator(new TranslationTranslator())
|
||||
.optEngine("PaddlePaddle") // Use PyTorch engine
|
||||
.optModelName("inference")
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.optProgress(new ProgressBar())
|
||||
.build();
|
||||
|
||||
|
@ -23,7 +23,7 @@ public final class Lac {
|
||||
.optTranslator(new LacTranslator())
|
||||
.optProgress(new ProgressBar())
|
||||
.optEngine("PaddlePaddle") // Use PaddlePaddle engine
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.build();
|
||||
|
||||
return criteria;
|
||||
|
@ -23,7 +23,7 @@ public final class SentaBilstm {
|
||||
// .optModelUrls("/Users/calvin/model/senta_bilstm/")
|
||||
.optTranslator(new SentaTranslator())
|
||||
.optEngine("PaddlePaddle") // Use PyTorch engine
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.optProgress(new ProgressBar())
|
||||
.build();
|
||||
|
||||
|
@ -24,7 +24,7 @@ public final class SentaLstm {
|
||||
// .optModelUrls("/Users/calvin/model/senta_lstm/")
|
||||
.optTranslator(new SentaTranslator())
|
||||
.optEngine("PaddlePaddle") // Use PyTorch engine
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.optProgress(new ProgressBar())
|
||||
.build();
|
||||
|
||||
|
@ -23,7 +23,7 @@ public final class Lac {
|
||||
.optTranslator(new LacTranslator())
|
||||
.optProgress(new ProgressBar())
|
||||
.optEngine("PaddlePaddle") // Use PaddlePaddle engine
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.build();
|
||||
|
||||
return criteria;
|
||||
|
@ -24,7 +24,7 @@ public final class SentaTextCnn {
|
||||
// .optModelUrls("/Users/calvin/model/senta_textcnn/")
|
||||
.optTranslator(new SentaTranslator())
|
||||
.optEngine("PaddlePaddle") // Use PyTorch engine
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.optProgress(new ProgressBar())
|
||||
.build();
|
||||
|
||||
|
@ -24,7 +24,7 @@ public final class ReviewDetection {
|
||||
// .optModelUrls("/Users/calvin/model/review_detection_lstm/")
|
||||
.optTranslator(new ReviewTranslator())
|
||||
.optEngine("PaddlePaddle") // Use PyTorch engine
|
||||
.optDevice(Device.cpu())
|
||||
// .optDevice(Device.cpu())
|
||||
.optProgress(new ProgressBar())
|
||||
.build();
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user