Skip to content

GitLab

"git@developer.sourcefind.cn:hehl2/torchaudio.git" did not exist on "479e666bd32c408f2833206f76d63e3cf6c2c700"
Commit 042c9aaa authored by Toby Boyd's avatar Toby Boyd Committed by Shining Sun
Browse files

Ncf metric tweaks (#6633) 

* Use tf.image.resize_with_crop_or_pad

* exp_per_second and hr_at_10
parent 5c37e69c
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
official/recommendation/ncf_keras_benchmark.py
View file @ 042c9aaa
...@@ -105,8 +105,8 @@ class KerasNCFRealData(KerasNCFBenchmarkBase): ...@@ -105,8 +105,8 @@ class KerasNCFRealData(KerasNCFBenchmarkBase):
def _extract_benchmark_report_extras(self, stats): def _extract_benchmark_report_extras(self, stats):
extras = {} extras = {}
extras['train_loss'] = stats['loss'] extras['train_loss'] = stats['loss']
extras['eval_hit_rate'] = stats['eval_hit_rate'] extras['hr_at_10'] = stats['eval_hit_rate']
extras['examples_per_second'] = stats['avg_exp_per_second'] extras['exp_per_second'] = stats['avg_exp_per_second']
return extras return extras
def benchmark_1_gpu(self): def benchmark_1_gpu(self):
...@@ -148,7 +148,7 @@ class KerasNCFSyntheticData(KerasNCFBenchmarkBase): ...@@ -148,7 +148,7 @@ class KerasNCFSyntheticData(KerasNCFBenchmarkBase):
def _extract_benchmark_report_extras(self, stats): def _extract_benchmark_report_extras(self, stats):
extras = {} extras = {}
extras['examples_per_second'] = stats['avg_exp_per_second'] extras['exp_per_second'] = stats['avg_exp_per_second']
return extras return extras
def benchmark_1_gpu(self): def benchmark_1_gpu(self):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment