Import json in train utilities

train_util.py

@@ -4,6 +4,7 @@ import logging
 import sys
 import time
 import csv
+import json
 from datetime import datetime
 from pathlib import Path
 from typing import Any, Dict, Tuple