What type of PR is this?

Uncomment only one /kind <> line, hit enter to put that in a new line, and remove leading whitespaces from that line:

/kind bug
/kind task
/kind feature
Still work in progress

What does this PR do / why do we need it:
Test resnet train
GLOG_v=2 pytest tests/perf_test/test_resnet_train.py -s 2>&1 |tee log.txt

Test bert infer, temp change the bert script for infer.
GLOG_v=2 pytest tests/perf_test/bert/test_bert_train.py::test_bert_train -s 2>&1 |tee log.txt
Which issue(s) this PR fixes:

Fixes #

Special notes for your reviewers: