def test_resnet18_ipex(self): resnet18 = vision.resnet18(pretrained=False, include_top=False, freeze=True) train_with_linear_top_layer(resnet18, batch_size, num_workers, data_dir, use_orca_lite_trainer=True)
def test_shufflenet_ipex(self): shufflenet = vision.shufflenet_v2_x1_0( pretrained=False, include_top=False, freeze=True) train_with_linear_top_layer( shufflenet, batch_size, num_workers, data_dir, accelerator=IPEXAccelerator())
def test_mobilenet_v3_small_ipex(self): mobilenet = vision.mobilenet_v3_small( pretrained=False, include_top=False, freeze=True) train_with_linear_top_layer( mobilenet, batch_size, num_workers, data_dir, accelerator=IPEXAccelerator())
def test_resnet50_ipex(self): resnet50 = vision.resnet50( pretrained=False, include_top=False, freeze=True) train_with_linear_top_layer( resnet50, batch_size, num_workers, data_dir, accelerator=IPEXAccelerator())
def test_mobilenet_v2(self): mobilenet = vision.mobilenet_v2( pretrained=False, include_top=False, freeze=True) train_with_linear_top_layer( mobilenet, batch_size, num_workers, data_dir)
def test_resnet34(self): resnet34 = vision.resnet34( pretrained=False, include_top=False, freeze=True) train_with_linear_top_layer( resnet34, batch_size, num_workers, data_dir)