azaion/annotations
1
0
Fork 0
mirror of https://github.com/azaion/annotations.git synced 2026-04-22 10:16:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

write logs for inference and loader to file

Browse Source
This commit is contained in:
Alex Bezdieniezhnykh
2025-06-14 16:08:32 +03:00
parent 8aa2f563a4
commit 6f297c4ebf
30 changed files with 218 additions and 140 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Azaion.Inference/tensorrt_engine.pyx
+2 -2
View File
@@ -103,11 +103,11 @@ cdef class TensorRTEngine(InferenceEngine):
constants.log('Converting to supported fp16')
config.set_flag(trt.BuilderFlag.FP16)
else:
print('Converting to supported fp32. (fp16 is not supported)')
constants.log('Converting to supported fp32. (fp16 is not supported)')
plan = builder.build_serialized_network(network, config)
if plan is None:
print('Conversion failed.')
constants.logerror('Conversion failed.')
return None
constants.log('conversion done!')
return bytes(plan)