kishwarshafin/helen

helen polish UserWarning on batch size

lloydlowUOA opened this issue · 0 comments

I am running helen polish and received below warning message. Is this a harmless warning or I should do something about it?

INFO: POLISH MODULE SELECTED
INFO: RUN-ID: 04112022_102154
INFO: PREDICTION OUTPUT DIRECTORY: /HELEN/predictions_04112022_102154
INFO: CALL CONSENSUS STARTING
INFO: OUTPUT FILE: /HELEN/predictions_04112022_102154/output_AngusONTpolish.fa
INFO: MODEL LOADING TO ONNX
INFO: SAVING MODEL TO ONNX
/opt/conda/lib/python3.7/site-packages/torch/onnx/symbolic_opset9.py:1436: UserWarning: Exporting a model to ONNX with a batch_size other than 1, with a variable lenght with GRU can cause an error when running the ONNX model with a different batch size. Make sure to save the model with a batch size of 1, or define the initial states (h0/c0) as inputs of the model.
"or define the initial states (h0/c0) as inputs of the model. ")
INFO: TORCH THREADS SET TO: 4.