[CodeClean] check null before using handle

Code clean, check null pointer of train-offloading handle before using it.

Signed-off-by: Jaeyun Jung <[email protected]>