Skip to content

Commit

Permalink
#52 fixed: used simplified API, passing only labels count.
Browse files Browse the repository at this point in the history
nicolay-r committed Jan 8, 2023
1 parent b6889ff commit 861066b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion arelight/pipelines/items/inference_bert.py
Original file line number Diff line number Diff line change
@@ -43,7 +43,7 @@ def __init__(self, bert_config_file, model_checkpoint_path, vocab_filepath, samp

self.__writer = predict_writer
self.__data_type = data_type
self.__labels_scaler = labels_scaler
self.__labels_count = labels_count
self.__predict_provider = BasePredictProvider()
self.__samples_io = samples_io
self.__batch_size = batch_size

0 comments on commit 861066b

Please sign in to comment.