search

liaoyuhua / tempo-pytorch

Reproduction of the paper "TEMPO: Prompt-based Generative Pre-trained Transformer for Time Series Forecasting"
MIT License
50 stars 1 forks source link

loss of prompt pool #1

Open JasonStraka opened 6 months ago

JasonStraka commented 6 months ago

作者你好， 复现的工作很出色，但是prompt pool的实现方法中并没有计算key的loss，与 https://github.com/JH-LEE-KR/l2p-pytorch/blob/main/prompt.py 中官方实现有一定差别。

liaoyuhua commented 6 months ago

稍后会优化下代码，谢谢提醒。