python画建筑_Python建筑网络的一个简单分类实例,pytorch,构建,示例

# encoding:utf-8

import pandas as pd

import torch

import torch.nn as nn

import torch.nn.functional as F

import torch.backends.cudnn as cudnn

import torch.optim as optim

from torch.utils.data import DataLoader

from torch.autograd import Variable

import os

os.environ['CUDA_LAUNCH_BLOCKING'] = "1"

#导入数据

train_data = pd.read_csv('train_data.csv')

test_data = pd.read_csv('test_data.csv')

#加速

torch.cuda.set_device(0) # 设置当前设备

cudnn.benchmark = True # 加速计算

cudnn.enabled = True # cuDNN是一个GPU加速深层神经网络原语库，开启cudnn

#转换数据类型为torch.tensor

train_data = torch.from_numpy(train_data.values)

test_data = torch.from_numpy(test_data.values)

print(train_data.size())

print(test_data.size())

#使用随机数据测试模型是否正常跑得