azaion/ai-training
1
0
Fork 0
mirror of https://github.com/azaion/ai-training.git synced 2026-04-23 00:56:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix tensor rt engine

Browse Source
This commit is contained in:
zxsanny
2025-03-28 14:50:43 +02:00
committed by Alex Bezdieniezhnykh
parent 5b89a21b36
commit 06a23525a6
16 changed files with 272 additions and 94 deletions
Download Patch File Download Diff File Expand all files Collapse all files
constants.py
+6 -2
View File
@@ -1,5 +1,4 @@
from os import path
from dto.annotationClass import AnnotationClass
azaion = '/azaion'
prefix = 'azaion-'
@@ -26,7 +25,6 @@ datasets_dir = path.join(azaion, 'datasets')
models_dir = path.join(azaion, 'models')
annotation_classes = AnnotationClass.read_json()
date_format = '%Y-%m-%d'
checkpoint_file = 'checkpoint.txt'
checkpoint_date_format = '%Y-%m-%d %H:%M:%S'
@@ -35,3 +33,9 @@ CONFIG_FILE = 'config.yaml'
ANNOTATIONS_QUEUE = 'azaion-annotations'
ANNOTATIONS_CONFIRMED_QUEUE = 'azaion-annotations-confirm'
OFFSET_FILE = 'offset.yaml'
AI_ONNX_MODEL_FILE_BIG = "azaion.onnx.big"
AI_ONNX_MODEL_FILE_SMALL = "azaion.onnx.small"
AI_TENSOR_MODEL_FILE_BIG = "azaion.engine.big"
AI_TENSOR_MODEL_FILE_SMALL = "azaion.engine.small"