update

Author: linjianma

recommendation/dlrm_v3/checkpoint.py

@@ -22,7 +22,7 @@
 import gin
 
 import torch
-from generative_recommenders.dlrm_v3.utils import MetricsLogger
+from utils import MetricsLogger
 from torch.distributed.checkpoint.stateful import Stateful
 from torch.optim.optimizer import Optimizer
 from torchrec.distributed.types import ShardedTensor

recommendation/dlrm_v3/data_producer.py

@@ -21,7 +21,7 @@
 from typing import List, Optional, Tuple
 
 import torch
-from generative_recommenders.dlrm_v3.datasets.dataset import Dataset, Samples
+from datasets.dataset import Dataset, Samples
 
 logging.basicConfig(level=logging.INFO)
 logger: logging.Logger = logging.getLogger("data_producer")