comment

train.py

@@ -127,7 +127,7 @@ def main_train(model):
     print(' Testing model finished! ')
 
     # ###################################################
-    # ارزیابی مدل آموزش دیده
+    # ارزیابی مدل آموزش دیده با متد evaluate_model از ماژول do_evaluate
     try:
         from evaluate_model import do_evaluate
         print(' Try to evaluating the trained model! ')
@@ -164,7 +164,7 @@ if __name__ == "__main__":
     # model = 'HooshvareLab/bert-fa-base-uncased-ner-peyma'
     # main_train(model)
     
-    # iterate models to train
+    # iterate models to train based on dataset
     for model in models:
         if model == '':
             continue
@@ -175,8 +175,9 @@ if __name__ == "__main__":
             
             if result:
                 print(f'TRAIN **{model}** Finished successfully')
-        except:
+        except Exception as error:
             print(f" !!! TRAIN **{model}** Model ERROR !!! ")
+            print(f"Error message: {error}")
     print('All Models Training Process Finished!')