search

wzh9969 / contrastive-htc

This repository implements a contrastive learning model for hierarchical text classification. This work has been accepted as the long paper "Incorporating Hierarchy into Text Encoder: a Contrastive Learning Approach for Hierarchical Text Classification" in ACL 2022.
MIT License
131 stars 30 forks source link

训练时长 #5

Closed YuboHe closed 2 years ago

YuboHe commented 2 years ago

您好，感觉网络结构还蛮复杂的，不知道WOS数据集，训练到论文中的表现需要多少个epoch呀，目前我用V100, 33min一轮，感觉时间好长

wzh9969 commented 2 years ago

log里显示一共20个epoch，我们使用单卡3090单个epoch的训练时间是20min左右

YuboHe commented 2 years ago

谢谢回复