Is there a better way to structure this code with inheritance?

Question

My code is organised as follows:

run_experiment.py
base_model.py
models/
  model_1.py
  model_2.py
  ...

Each model inherits from base_