(engine, path)
| 131 | |
| 132 | |
| 133 | def serialize_engine(engine, path): |
| 134 | logger.info(f"Serializing engine to {path}...") |
| 135 | tik = time.time() |
| 136 | if isinstance(engine, trt.ICudaEngine): |
| 137 | engine = engine.serialize() |
| 138 | with open(path, "wb") as f: |
| 139 | f.write(engine) |
| 140 | tok = time.time() |
| 141 | t = time.strftime("%H:%M:%S", time.gmtime(tok - tik)) |
| 142 | logger.info(f"Engine serialized. Total time: {t}") |
| 143 | |
| 144 | |
| 145 | def deserialize_engine(path): |