beit_finetuning/modeling_finetune.py (1 line): - line 197: # FIXME look at relaxing size constraints main_linear.py (1 line): - line 354: # TODO: this should also be done with the ProgressMeter