|
|
@@ -120,7 +120,7 @@ def test_resnet50_quant(): |
|
|
dataset_sink_mode=False) |
|
|
dataset_sink_mode=False) |
|
|
print("============== End Training ==============") |
|
|
print("============== End Training ==============") |
|
|
|
|
|
|
|
|
expect_avg_step_loss = 2.40 |
|
|
|
|
|
|
|
|
expect_avg_step_loss = 2.50 |
|
|
avg_step_loss = np.mean(np.array(monitor.losses)) |
|
|
avg_step_loss = np.mean(np.array(monitor.losses)) |
|
|
|
|
|
|
|
|
print("average step loss:{}".format(avg_step_loss)) |
|
|
print("average step loss:{}".format(avg_step_loss)) |
|
|
|