forked from mindspore/mindspore
36 lines
999 B
Python
36 lines
999 B
Python
import os
|
|
import mindspore as ms
|
|
import numpy as np
|
|
|
|
class Exp_Basic(object):
|
|
def __init__(self, args):
|
|
self.args = args
|
|
self.device = self._acquire_device()
|
|
self.model = self._build_model()
|
|
|
|
def _build_model(self):
|
|
raise NotImplementedError("This method should be overridden in subclasses.")
|
|
return None
|
|
|
|
def _acquire_device(self):
|
|
if self.args.use_gpu:
|
|
os.environ["DEVICE_ID"] = str(self.args.gpu) if not self.args.use_multi_gpu else self.args.devices
|
|
device = ms.context.PYNATIVE_MODE # MindSpore uses context for device management
|
|
print('Use GPU: {}'.format(self.args.gpu))
|
|
else:
|
|
ms.context.set_context(mode=ms.context.PYNATIVE_MODE, device_target="CPU")
|
|
device = "CPU"
|
|
print('Use CPU')
|
|
return device
|
|
|
|
def _get_data(self):
|
|
pass
|
|
|
|
def vali(self):
|
|
pass
|
|
|
|
def train(self):
|
|
pass
|
|
|
|
def test(self):
|
|
pass |