From 2a55a5f561912778bfc10c50e213f53bcfbe27c8 Mon Sep 17 00:00:00 2001
From: Vincent <hailinfufu@outlook.com>
Date: Thu, 2 Nov 2023 18:18:43 +0800
Subject: [PATCH] enrich the hint information

---
 deepray/core/base_trainer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/deepray/core/base_trainer.py b/deepray/core/base_trainer.py
index 2f54ea7..0c0be13 100644
--- a/deepray/core/base_trainer.py
+++ b/deepray/core/base_trainer.py
@@ -222,7 +222,7 @@ def __init__(
       if len(main_keys) == 1:
         self._model = model
       else:
-        raise ValueError("Must set one model with key contains \"main\"")
+        raise ValueError(f"Must set only one model with key contains \"main\", found {main_keys}.")
     elif isinstance(model, tf.keras.Model):
       self._model = {"main_model": model}
     else: