✍️ update contextnet scripts

nglehuy · nglehuy · commit b2bfd92baf78 · 2021-02-20T19:16:21.000+07:00
diff --git a/examples/contextnet/train_tpu_keras_subword_contextnet.py b/examples/contextnet/train_tpu_keras_subword_contextnet.py
@@ -115,7 +115,6 @@
 
     contextnet.compile(
         optimizer=optimizer,
-        experimental_steps_per_execution=args.spx,
         global_batch_size=global_batch_size,
         blank=text_featurizer.blank
     )
@@ -132,5 +131,4 @@
     contextnet.fit(
         train_data_loader, epochs=config.learning_config.running_config.num_epochs,
         validation_data=eval_data_loader, callbacks=callbacks,
-        steps_per_epoch=train_dataset.total_steps, validation_steps=eval_dataset.total_steps
     )

Original file line number	Diff line number	Diff line change
`@@ -115,7 +115,6 @@`
`115`	`115`
`116`	`116`	`contextnet.compile(`
`117`	`117`	`optimizer=optimizer,`
`118`		`- experimental_steps_per_execution=args.spx,`
`119`	`118`	`global_batch_size=global_batch_size,`
`120`	`119`	`blank=text_featurizer.blank`
`121`	`120`	`)`
`@@ -132,5 +131,4 @@`
`132`	`131`	`contextnet.fit(`
`133`	`132`	`train_data_loader, epochs=config.learning_config.running_config.num_epochs,`
`134`	`133`	`validation_data=eval_data_loader, callbacks=callbacks,`
`135`		`- steps_per_epoch=train_dataset.total_steps, validation_steps=eval_dataset.total_steps`
`136`	`134`	`)`