I'm saying that this should really be a function like

    def train(
      data: File,
      arch='resnet18',
      workers=4,
      epochs=90,
      start_epoch=0,
      batch_size=256,
      learning_rate=0.9,
      momentum=1e-4,
  ):
    ...

that could be exposed as a cli if __name__ is main but otherwise can be imported in a notebook or another script.