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

train is ready

Browse Source 
manual_run reuses train's export
add current_model to constants
This commit is contained in:
Alex Bezdieniezhnykh
2025-05-31 18:41:10 +03:00
parent 0bd5bd6d27
commit 538dc8efa9
4 changed files with 18 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files
constants.py
+3 -6
View File
@@ -34,12 +34,9 @@ TXT_EXT = '.txt'
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"
SMALL_SIZE_KB = 3
CDN_CONFIG = 'cdn.yaml'
MODELS_FOLDER = 'models'
CURRENT_PT_MODEL = path.join(models_dir, f'{prefix[:-1]}.pt')
CURRENT_ONNX_MODEL = path.join(models_dir, f'{prefix[:-1]}.onnx')