def test_resetnet50_build(): X = me.tensor() Y = me.tensor() X.set_dtype(dtype.float32) Y.set_dtype(dtype.float32) network = resnet50() model = Model(network=network, loss_fn=loss_func, optimizer=optimizer)
def test_tensor_add(): X = me.tensor() Y = me.tensor() X.set_dtype(dtype.float32) Y.set_dtype(dtype.float32) X = me.tensor(np.ones([2, 3])) Y = me.tensor(np.ones([2, 3])) sum = add(X, Y) print("test tensor add") return sum