想问一下,STAR: 基于Transformer的行人轨迹预测模型,这篇论文的可视化是怎么实现的,如果可以,能不能分享一下可视化的代码,谢谢!下面是代码中保存的参数
def save_model(self, epoch):
model_path = self.args.save_dir + '/' + self.args.train_model + '/' + self.args.train_model + '_' + \
str(epoch) + '.tar'
torch.save({
'epoch': epoch,
'state_dict': self.net.state_dict(),
'optimizer_state_dict': self.optimizer.state_dict()
}, model_path)