diff --git a/trainers/dcn_trainer.py b/trainers/dcn_trainer.py index f0067c4..4819b6f 100644 --- a/trainers/dcn_trainer.py +++ b/trainers/dcn_trainer.py @@ -1,3 +1,5 @@ +import wandb + import numpy as np import pandas as pd from tqdm import tqdm diff --git a/trainers/mf_trainer.py b/trainers/mf_trainer.py index f39da10..cf8f6a5 100644 --- a/trainers/mf_trainer.py +++ b/trainers/mf_trainer.py @@ -1,3 +1,5 @@ +import wandb + import numpy as np import pandas as pd from tqdm import tqdm