learning/annotated_deep_learning_paper_implementations
1
0
Fork 0
mirror of https://github.com/labmlai/annotated_deep_learning_paper_implementations.git synced 2025-08-26 08:41:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

cleanup log activations

Browse Source
This commit is contained in:
Varuna Jayasiri
2025-07-20 09:10:05 +05:30
parent a713c92b82
commit 5eecda7e28
12 changed files with 68 additions and 136 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
labml_nn/optimizers/mnist_experiment.py
View File

@ -71,9 +71,8 @@ class Configs(MNISTConfigs, TrainValidConfigs):
if self.mode.is_train:
tracker.add_global_step(len(data))
# Run the model and specify whether to log the activations
with self.mode.update(is_log_activations=batch_idx.is_last):
output = self.model(data)
# Run the model
output = self.model(data)
# Calculate the loss
loss = self.loss_func(output, target)