# trainingsetor testing set,valsetmnistdata=MovingMNISTdataset("./data/mnist_test_seq.npy")train_size=int(0.8*len(mnistdata))test_size=len(mnistdata)-train_size torch.manual_seed(torch.initial_seed())train_dataset,test_dataset=random_split(mnistdata,[train_size,test_size])num_train=len(tr...
在 MovingMNIST数据集上训练 ConvLSTM 模型所需的 epoch 数因模型架构、学习率、批量大小等各种因素而异。
ConvLSTM-Moving-mnist Using convlstm to prediction moving mnist dataset. run main.py for train 1、network structure 使用其中一个序列,迭代训练,收敛。 前10帧--第11帧,...,第10-19帧预测第20帧。 输入mnist序列 三层conv 一层BasicConvLstmCel 三层conv flatten(1,10,64,64,1) (1,10,64,64,16)...
mnistdata = MovingMNISTdataset("./data/mnist_test_seq.npy") train_size = int(0.8 * len(mnistdata)) test_size = len(mnistdata) - train_size torch.manual_seed(torch.initial_seed()) train_dataset, test_dataset = random_split(mnistdata, [train_size, test_size]) num_train = len(train...
但是是不是因为epoch不够这个我也不知道，也可能是代码有问题，比如数据读取，loss计算都有可能。我可能...