Skip to content

Instantly share code, notes, and snippets.

@jcreinhold

jcreinhold/train.py

Created January 3, 2019 15:25
Show Gist options
  • Save jcreinhold/4cf62e24363c185911d1e24b5c4e09e5 to your computer and use it in GitHub Desktop.
Save jcreinhold/4cf62e24363c185911d1e24b5c4e09e5 to your computer and use it in GitHub Desktop.
Download ZIP
training of 3D resnet
Raw
train.py
loss = nn.MSELoss()
learner = faiv.Learner(idb, model, loss_func=loss)
learner.lr_find()
learner.recorder.plot() # pick learning rate based on this plot
cbs = [faiv.callbacks.CSVLogger(learner, 'history')]
learner.fit_one_cycle(100, 1e-2, callbacks=cbs)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment