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-02 21:40:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

♻️ tracker for tracking models

Browse Source
This commit is contained in:
Varuna Jayasiri
2020-12-02 13:58:24 +05:30
parent 5238b5d432
commit 6f6b185aae
4 changed files with 7 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
labml_nn/transformers/knn/train_model.py
View File

@ -10,7 +10,6 @@ import torch
import torch.nn as nn
from torchtext.data.utils import get_tokenizer
import labml.utils.pytorch as pytorch_utils
from labml import lab, experiment, monit, tracker, logger
from labml.configs import option
from labml.logger import Text
@ -174,7 +173,7 @@ class Configs(SimpleTrainValidConfigs):
loss.backward()
self.optimizer.step()
if batch_idx.is_last:
pytorch_utils.store_model_indicators(self.model)
tracker.add('model', self.model)
self.optimizer.zero_grad()
tracker.save()