From ea5db513ffad744fda950d6f0980739d13749811 Mon Sep 17 00:00:00 2001 From: moe chabot Date: Mon, 15 Sep 2025 10:06:00 -0400 Subject: [PATCH] Update solution.py --- questions/160_mixed_precision_training/solution.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/questions/160_mixed_precision_training/solution.py b/questions/160_mixed_precision_training/solution.py index d4b97ad3..76963bec 100644 --- a/questions/160_mixed_precision_training/solution.py +++ b/questions/160_mixed_precision_training/solution.py @@ -1,6 +1,5 @@ import numpy as np - class MixedPrecision: def __init__(self, loss_scale=1024.0): self.loss_scale = loss_scale @@ -15,9 +14,8 @@ def forward(self, weights, inputs, targets): predictions = np.dot(inputs_fp16, weights_fp16) loss = np.mean((targets_fp16 - predictions) ** 2) - # Scale loss and convert back to float32 - scaled_loss = loss.astype(np.float32) * self.loss_scale - + # Scale loss and convert back to float32 (Python float) + scaled_loss = float(loss) * self.loss_scale return scaled_loss def backward(self, gradients):