def create_network(name, *args, **kwargs): if name == 'ghostnet': return ghostnet_1x(*args, **kwargs) if name == 'ghostnet_nose': return ghostnet_nose_1x(*args, **kwargs) if name == 'ghostnet-600': return ghostnet_600m(*args, **kwargs) raise NotImplementedError(f"{name} is not implemented in the repo")
device_target="Ascend", device_id=device_id, save_graphs=False) elif args_opt.platform == "GPU": config_platform = config_gpu context.set_context(mode=context.GRAPH_MODE, device_target="GPU", save_graphs=False) else: raise ValueError("Unsupport platform.") loss = nn.SoftmaxCrossEntropyWithLogits(is_grad=False, sparse=True, reduction='mean') net = ghostnet_1x(num_classes=config_platform.num_classes) if args_opt.platform == "Ascend": net.to_float(mstype.float16) for _, cell in net.cells_and_names(): if isinstance(cell, nn.Dense): cell.to_float(mstype.float32) dataset = create_dataset(dataset_path=args_opt.dataset_path, do_train=False, config=config_platform, platform=args_opt.platform, batch_size=config_platform.batch_size) step_size = dataset.get_dataset_size() if args_opt.checkpoint_path: