Stop passing None to compile() in TF examples (#29597)

* Fix examples to stop passing None to compile(), rework example invocation for run_text_classification.py

* Add Amy's fix
This commit is contained in:
Matt
2024-03-12 12:22:29 +00:00
committed by GitHub
parent 73efe896df
commit 81ec8028f9
8 changed files with 12 additions and 8 deletions

View File

@@ -706,7 +706,8 @@ def main():
model.compile(optimizer=optimizer, jit_compile=training_args.xla, metrics=["accuracy"])
else:
model.compile(optimizer=None, jit_compile=training_args.xla, metrics=["accuracy"])
# Optimizer doesn't matter as it won't be used anyway
model.compile(optimizer="sgd", jit_compile=training_args.xla, metrics=["accuracy"])
training_dataset = None
if training_args.do_eval: