From 855a456074fa5e2a875c0ca61ced8f2d7a1a89a7 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=A4=A9=E9=82=91?= Date: Wed, 30 Oct 2024 15:40:47 +0800 Subject: [PATCH] pyfg use avx512 according to cpuinfo --- docs/source/models/loss.md | 2 +- requirements/runtime.txt | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/docs/source/models/loss.md b/docs/source/models/loss.md index 03e219d..0bf1fb0 100644 --- a/docs/source/models/loss.md +++ b/docs/source/models/loss.md @@ -28,7 +28,7 @@ model_config { ``` model_config { losses { - SoftmaxCrossEntropy { + softmax_cross_entropy { } } } diff --git a/requirements/runtime.txt b/requirements/runtime.txt index ed9ebdd..cbc34db 100644 --- a/requirements/runtime.txt +++ b/requirements/runtime.txt @@ -7,8 +7,8 @@ graphlearn @ https://tzrec.oss-cn-beijing.aliyuncs.com/third_party/graphlearn-1. graphlearn @ https://tzrec.oss-cn-beijing.aliyuncs.com/third_party/graphlearn-1.3.0-cp310-cp310-linux_x86_64.whl ; python_version=="3.10" grpcio-tools<1.63.0 pandas -pyfg @ https://tzrec.oss-cn-beijing.aliyuncs.com/third_party/pyfg-0.3.2-cp311-cp311-linux_x86_64.whl ; python_version=="3.11" -pyfg @ https://tzrec.oss-cn-beijing.aliyuncs.com/third_party/pyfg-0.3.2-cp310-cp310-linux_x86_64.whl ; python_version=="3.10" +pyfg @ https://tzrec.oss-cn-beijing.aliyuncs.com/third_party/pyfg-0.3.4-cp311-cp311-linux_x86_64.whl ; python_version=="3.11" +pyfg @ https://tzrec.oss-cn-beijing.aliyuncs.com/third_party/pyfg-0.3.4-cp310-cp310-linux_x86_64.whl ; python_version=="3.10" scikit-learn tensorboard torch==2.5.0