pytorch使用bert进行多分类

背景 之前虽然会用keras来搭建模型,但后来发现用keras修改具体的底层细节实现比较麻烦。又不想用tensorflow写那么多代码,所以近

keras_tf2数据并行

背景 一直都想去搭一个并行模型,即使用多gpu共同计算的模型。今天终于把keras+tf2下的数据并行模式调通啦~~ 关键点 1 2 3 4 5 6 import os os.environ["CUDA_VISIBLE_DEVICES"] =

股票量化监控系统

背景 一直都想去搭一个量化交易的平台。从一开始搭建的 與情预测股票系统 ,锻炼了爬虫技术、数据库技术和机器学习技术;到后来的 股票价格时序分解系统 锻

使用tfrecord构建数据集

背景 最近在打比赛,听说tfrecord的格式可以以二进制的方式高效存储数据,并利用protobuf协议读取数据,所以就想着把训练代码改成tf

获取模型的中间层输出

背景 最近在疯狂搭模型,突然有人来问我要我模型的中间层输出的embedding,于是,我就研究了一下怎么获取模型的中间层输出。 代码实现 1 2 3 4