Merge pull request #2053 from NirajanBekoju/nirajan

fix: argument value in automatic selection of model
RUCAIBox · Aug 28, 2024 · 29bbe4c · 29bbe4c
2 parents 14430b3 + e6ab5c3
commit 29bbe4c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/source/user_guide/usage/use_modules.rst b/docs/source/user_guide/usage/use_modules.rst
@@ -127,7 +127,7 @@ trainer class through the model name.
         ...
 
         # model loading and initialization
-        model = get_model(config['model'])(config, train_data).to(config['device'])
+        model = get_model(config['model'])(config, train_data.dataset).to(config['device'])
 
         # trainer loading and initialization
         trainer = get_trainer(config['MODEL_TYPE'], config['model'])(config, model)