项目讲解:Transformer进行船舶轨迹预测 -完整代码数据视频简介链接自取_哔哩哔哩_bilibili
本博客付完整的代码数据
项目代码:
import argparse
from dataread import data_train,data_test
import torch
import numpy as np
from transformer.Models import Transformer
import os
from torch import optim
from transformer.Optim import ScheduledOptim
from torch.nn import functional as F
import random
from tqdm import tqdm
from matplotlib import pyplot as plt
os.environ['CUDA