2 ms·
I'm saying that this should really be a function like def train( data: File, arch='resnet18', workers=4, epochs=90, start_epo
by m_ke 8y ago
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.