fix: change argument value to train_data.dataset for automatic select…

…ion of model
RUCAIBox · May 30, 2024 · e6ab5c3 · e6ab5c3
1 parent 2b6e209
commit e6ab5c3
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)