nvidia merlin

raj200501 · raj200501 · commit e5f5e4f6da84 · 2024-02-21T10:43:38.000-05:00
diff --git a/model.py b/model.py
@@ -20,6 +20,7 @@
 # Assuming 'train.schema' is defined based on the 'train' Dataset
 # The schema should describe the structure of the input data for the model.
 
+#merlin model NVIDIA
 model = mm.DLRMModel(
     train.schema,  # The schema should be obtained from the 'train' dataset
     embedding_dim=64,
