MCPcopy Create free account
hub / github.com/NVIDIA/TensorRT / compile_model

Function compile_model

tools/tensorflow-quantization/examples/utils_finetuning.py:73–78  ·  view source on GitHub ↗
(model)

Source from the content-addressed store, hash-verified

71
72
73def compile_model(model):
74 model.compile(
75 optimizer="sgd",
76 loss=tf.keras.losses.SparseCategoricalCrossentropy(),
77 metrics=["accuracy"],
78 )
79
80
81def fine_tune(

Callers 3

mainFunction · 0.90
mainFunction · 0.90
mainFunction · 0.90

Calls

no outgoing calls

Tested by

no test coverage detected