baiyfbupt

import torch
import torch.nn as nn
import paddle
import paddle.fluid as fluid
import unittest
import time
import numpy as np

device_type = "cpu"
#device_type = "cuda:0"