naver/sqlova

assert problem

Opened this issue · 1 comments

sqlova/utils/utils_wikisql.py line
def get_bert_output
len 728-735

`

        while len(input_ids1) < max_seq_length:
                    input_ids1.append(0)
                    input_mask1.append(0)
                    segment_ids1.append(0)

        assert len(input_ids1) == max_seq_length
        assert len(input_mask1) == max_seq_length
        assert len(segment_ids1) == max_seq_length

`
There will raise a error if --max_seq_leng is smaller than seq length.

you could cut the length to max_seq_leng, maybe can solve it?