search

princeton-nlp / SimCSE

[EMNLP 2021] SimCSE: Simple Contrastive Learning of Sentence Embeddings https://arxiv.org/abs/2104.08821
MIT License
3.41k stars 515 forks source link

How to add values into tensorboard recorders？ #119

Closed marscrazy closed 2 years ago

marscrazy commented 2 years ago

Thanks for your great works. It's a very simple and effective method to encode sentences. I try to record some values during the training process, such as the loss term. But I cannot find the code like the SummaryWriter in tensorboardx. Could you tell me which file or function I should look for?

gaotianyu1350 commented 2 years ago

Hi,

Our code is based on Huggingface's transformers, where they have already integrated tensorboard support in trainer.py.

marscrazy commented 2 years ago

Thanks.