浪子寒的策略空间

简单粗暴# tensorflow 内存不够的情况 适用于tensorflow 2.ximport tensorflow as tfos.environ["CUDA_VISIBLE_DEVICES"] = "0" # 指定显卡config = tf.compat.v1.ConfigProto(allow_soft_placement=True) # 开启软放置，OP放到CPU上了，为啥源码默认值是7？config.gpu_options.per_process_gpu_memory_fract

文章目录#模块导入，常量参数设定import torchimport torch.nn as nnimport torch.nn.functional as Fimport torch.optim as optimfrom torchvision import datasets, transformsfrom torchsummary import summaryBATCH_SIZE = 512EPOCH = 20DEIVCE = torch.device('cuda:0' if to

编程环境：python 3.6.8tensorflow 1.12.3点击下载离线包matplotlib 3.1.2numpy 1.17.4数据集说明：我所采用的数据集，是我自己构建的一个网络流量数据集，借鉴了Wei Wang等人端到端的思想，但是处理成的数据集却不同于他们的MNIST型数据集，而是采用的npy进行存储。由于只是用于测试模型搭建，该数据集仅包含了一部...