这是关于 PatternTransformer 的实验代码。 make_datasets:用来做数据预处理以及形成 datsets。 models:是 PatternTransformer 的网络框架。 utils:是在数据预处理过程中可能用到的函数,以及实验过程中的评估方法。 main.py: 帮助我们了解数据集的分布情况。 run_PT.py: 为程序运行的脚本,包含了参数的定义,模型的调用以及训练和测试的代码。
-
Notifications
You must be signed in to change notification settings - Fork 0
kcisgroup/PatternTransformer
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published