From e6ab5c3e7711018f83774a97aedbb90d14a11bea Mon Sep 17 00:00:00 2001 From: NirajanBekoju Date: Thu, 30 May 2024 13:53:32 +0545 Subject: [PATCH] fix: change argument value to train_data.dataset for automatic selection of model --- docs/source/user_guide/usage/use_modules.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/user_guide/usage/use_modules.rst b/docs/source/user_guide/usage/use_modules.rst index b98c29e10..e1f327488 100644 --- 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)