新闻分类大赛task5

最新推荐文章于 2022-04-22 12:56:26 发布

我是飞行模式

最新推荐文章于 2022-04-22 12:56:26 发布

阅读量94

点赞数

分类专栏：新闻分类比赛学习

本文链接：https://blog.csdn.net/u012302260/article/details/107722369

版权

新闻分类比赛学习专栏收录该内容

6 篇文章 0 订阅

订阅专栏

深度学习的文本分类TextCNN

import logging
import random

import numpy as np
import torch

logging.basicConfig(level=logging.INFO, format='%(asctime)-15s %(levelname)s: %(message)s')

# set seed 
seed = 666
random.seed(seed)
np.random.seed(seed)
torch.cuda.manual_seed(seed)
torch.manual_seed(seed)

# set cuda
gpu = 0
use_cuda = gpu >= 0 and torch.cuda.is_available()
if use_cuda:
    torch.cuda.set_device(gpu)
    device = torch.device("cuda", gpu)
else:
    device = torch.device("cpu")
logging.info("Use cuda: %s, gpu id: %d.", use_cuda, gpu)