浪子寒的策略空间

简单粗暴 # tensorflow 内存不够的情况 适用于tensorflow 2.x import tensorflow as tf os.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 torch import torch.nn as nn import torch.nn.functional as F import torch.optim as optim from torchvision import datasets, transforms from torchsummary import summary BATCH_SIZE = 512 EPOCH = 20 DEIVCE = torch.device('cuda:0' if to

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