Incorrect predictions during forward propagation

[bhqv]

Description

Issue with forward propagation. When I input data into the model for inference, the predictions seem to be incorrect or inconsistent .

Difficulty

• Intermediate
• 30 Points (FOR EACH CORRECTION)

File with Issue: dig.py

Expected Behavior

I expect the model to generate predictions that closely match the ground truth labels for the given input data.

Actual Behavior

The predictions generated by the model do not match the ground truth labels, indicating an issue with the forward propagation process.

HINT: Bringing clarity to complexity, some secret issues needs resolving