Update README.MD

anuaspriya · Jan 26, 2022 · 7cdf79b · 7cdf79b
1 parent 992878a
commit 7cdf79b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/README.MD b/README.MD
@@ -45,7 +45,8 @@ from pyabsa.functional import Trainer
 from autocuda import auto_cuda
 
 config = APCConfigManager.get_apc_config_english() # APC task
-dataset = '100.CustomDataset'
+dataset = '101.restaurant' 
+# dataset = '100.CustomDataset'
 Trainer(config=config,
         dataset=dataset,  # train set and test set will be automatically detected
         checkpoint_save_mode=1,