复现论文:PromptTA: Prompt-driven Text Adapter for Source-freeDomain Generalization

github:zhanghr2001/PromptTA: Source-free Domain Generalization

论文:[2409.14163] PromptTA: Prompt-driven Text Adapter for Source-free Domain Generalization

自己标注:PromptTA: Prompt-driven Text Adapter for Source-free Domain Generalizationhttps://github.com/Unchanged-Originality/Unchanged-Originality/blob/main/Prompt-driven%20Text%20Adapter%20for%20Source-free%20Domain%20Generalization.pdf

流程图:

这篇论文主要是受 Domain-Unified Prompt Representations for Source-Free Domain Generalization和PromptStyler: Prompt-driven Style Generation for Source-free Domain Generalization启发。

github:这两篇论文我读标记

复现结果:

|-----------|------|------|------------|-----------|
| | PACS | VLCS | OfficeHome | DomainNet |
| ResNet-50 | 94.0 | 83.3 | 74.1 | |
| ViT-B/16 | 97.4 | 84.9 | 82.9 | |

PACS:

ResNet-50:

A:94.4 94.4 94.9 平均:94.733

C:95.3 95.2 95.2 95.233

P:99.5 99.5 99.5 99.5

s:86.5 86.4 86.7 86.567

平均 top - 1 准确率94.0

ViT-B/16:

a:97.9 97.9 98 97.933

C: 99.0 99.1 99.1 99.067

P:99.9 99.9 99.9 99.9

s:92.4 93.0 93.6 93

平均 top - 1 准确率97.475

VLCS:

ResNet-50:

C:100 100 100 平均:100

L:70.9 70.8 69.5 70.4

P:88.4 87.4 87.2 87.667

S:75.8 74.8 74.8 75.133

平均 top - 1 准确率83.3

ViT-B/16:

C:100 100 100 平均:100

L:71.3 74.7 75 73.667

P:89.8 90.4 90.1 90.1

S:77.1 75.3 76.0 76.133

平均 top - 1 准确率84.9

OfficeHome

ResNet-50:

A:73.3 73.3 73.5平均:73.367

C:55.3 55.2 55.0 55.167

P:84.2 84.3 83.9 84.133

R: 84.1 84.1 83.9 84.033

平均 top - 1 准确率74.1

ViT-B/16:

A:81.5 81.7 81.7 平均:81.633

C:70.1 70.0 70.6 70.233

P:89.7 90.0 89.8 89.833

r:90.0 90.3 90.2 90.167

平均 top - 1 准确率82.9

报错:

按照github配置环境过程中没出现问题。

AttributeError: module 'torch.utils.data' has no attribute 'collate'

改了半天,发现是原作者的引用错了

Traceback (most recent call last):

File "train.py", line 10, in <module>

from trainers import *

File "/opt/data/private/promptta/trainers/init.py", line 4, in <module>

from .prompt_ta import PROMPT_TA

File "/opt/data/private/promptta/trainers/prompt_ta.py", line 12, in <module>

from torch.utils.data import *

AttributeError: module 'torch.utils.data' has no attribute 'collate'

把prompt_ta.py文件中的 from torch.utils.data import *注释掉,改成

复制代码
from torch.utils.data import DataLoader, Dataset
复制代码
from torch.utils.data import TensorDataset

然后就可以正常运行了。

相关推荐
程序员ken9 分钟前
深入理解大语言模型(8) 使用 LangChain 开发应用程序之上下文记忆
人工智能·python·语言模型·langchain
wazmlp00188736917 分钟前
第五次python作业
服务器·开发语言·python
尘缘浮梦19 分钟前
websockets简单例子1
开发语言·python
不懒不懒19 分钟前
【从零开始:PyTorch实现MNIST手写数字识别全流程解析】
人工智能·pytorch·python
helloworld也报错?20 分钟前
基于CrewAI创建一个简单的智能体
人工智能·python·vllm
啊阿狸不会拉杆33 分钟前
《机器学习导论》第 10 章-线性判别式
人工智能·python·算法·机器学习·numpy·lda·线性判别式
超龄超能程序猿34 分钟前
Python 反射入门实践
开发语言·python
玄同76542 分钟前
Python Random 模块深度解析:从基础 API 到 AI / 大模型工程化实践
人工智能·笔记·python·学习·算法·语言模型·llm
AIFarmer1 小时前
在EV3上运行Python语言——环境设置
python·ev3