Created by: heavengate
Function optimization
APIs
use paddle.get_default_dtype for vision datasets:
paddle.get_default_dtype