learning/annotated_deep_learning_paper_implementations
1
0
Fork 0
mirror of https://github.com/labmlai/annotated_deep_learning_paper_implementations.git synced 2025-11-03 22:09:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

remove labml_helpers dep

Browse Source
This commit is contained in:
Varuna Jayasiri
2025-07-20 08:56:03 +05:30
parent b1ba92c166
commit 5bdedcffec
39 changed files with 1356 additions and 71 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
labml_nn/uncertainty/evidence/experiment.py
View File

@ -18,8 +18,8 @@ import torch.utils.data
from labml import tracker, experiment
from labml.configs import option, calculate
from labml_helpers.schedule import Schedule, RelativePiecewise
from labml_helpers.train_valid import BatchIndex
from labml_nn.helpers.schedule import Schedule, RelativePiecewise
from labml_nn.helpers.trainer import BatchIndex
from labml_nn.experiments.mnist import MNISTConfigs
from labml_nn.uncertainty.evidence import KLDivergenceLoss, TrackStatistics, MaximumLikelihoodLoss, \
CrossEntropyBayesRisk, SquaredErrorBayesRisk
@ -178,7 +178,7 @@ def kl_div_coef(c: Configs):
### KL Divergence Loss Coefficient Schedule
"""
# Create a [relative piecewise schedule](https://docs.labml.ai/api/helpers.html#labml_helpers.schedule.Piecewise)
# Create a [relative piecewise schedule](../../helpers/schedule.html)
return RelativePiecewise(c.kl_div_coef_schedule, c.epochs * len(c.train_dataset))