Hi Yicong Cui
Issue seems to be in scoring script. Normally driver/amlbi_main.py issues point towards missing environment packages or runtime issue in loading model etc.
You can debug it by adding try catch blocks in crucial steps like model/data loading, saving to storage location etc.
First point of debugging should be in endpoint logs.
If you could share endpoint and environment logs in private message, we can debug further.
Attached Troubleshoot guide for batch endpoint for reference
Thank you.