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)