models/omnivore_model.py (1 line): - line 149: drop_path_rate=0.3, # TODO: set this based on the final models