From 9c7dbb9ca70858b870f74ecf595d3169f0cbc711 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Tue, 27 Sep 2022 23:11:06 +0200
Subject: Rename mapping to tokenizer

---
 training/run.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

(limited to 'training/run.py')

diff --git a/training/run.py b/training/run.py
index 68cedc7..99059d6 100644
--- a/training/run.py
+++ b/training/run.py
@@ -14,7 +14,6 @@ from pytorch_lightning import (
 from pytorch_lightning.loggers import LightningLoggerBase
 from torch import nn
 from torchinfo import summary
-
 import utils
 
 
@@ -39,7 +38,7 @@ def run(config: DictConfig) -> Optional[float]:
     model: LightningModule = hydra.utils.instantiate(
         config.model,
         network=network,
-        mapping=datamodule.mapping,
+        tokenizer=datamodule.tokenizer,
         loss_fn=loss_fn,
         optimizer_config=config.optimizer,
         lr_scheduler_config=config.lr_scheduler,
-- 
cgit v1.2.3-70-g09d2