Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
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
73
def
compile_model(model):
74
model.compile(
75
optimizer=
"sgd"
,
76
loss=tf.keras.losses.SparseCategoricalCrossentropy(),
77
metrics=[
"accuracy"
],
78
)
79
80
81
def
fine_tune(
Callers
3
main
Function · 0.90
main
Function · 0.90
main
Function · 0.90
Calls
no outgoing calls
Tested by
no test coverage detected