Skip to content

Conversation

@Baekpica
Copy link

@Baekpica Baekpica commented Oct 4, 2021

When BERT model calculate accuracy score in ./BERT/bert_trainer.py, current code adds all correct rows in for loop of data_loader and then divides by float(len(data_loader)).

But, len(data_loader) will return # of batches, not of all rows.

So I changed len(data_loader) to len(data_loader.dataset), which means total rows.

@Baekpica
Copy link
Author

Baekpica commented Oct 4, 2021

from
KakaoTalk_20211004_222854922

to
KakaoTalk_20211004_222951945

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant