Dean commited on
Commit
34a1202
1 Parent(s): fec2513

Seems like we are now using the correct format for fastai2. Still there is a strange bug where the signal is killed in training

Browse files
Files changed (1) hide show
  1. src/code/training.py +2 -4
src/code/training.py CHANGED
@@ -13,15 +13,13 @@ def get_y_fn(x):
13
 
14
  def create_data(data_path):
15
  fnames = get_files(data_path/'train', extensions='.jpg')
16
- data = SegmentationDataLoaders.from_label_func(data_path/'train', fnames=fnames, label_func=get_y_fn)
17
  return data
18
 
19
 
20
  def train(data):
21
- data.num_workers = 0
22
- data.bs = 1
23
  learner = unet_learner(data, resnet34, metrics=rmse, wd=1e-2, n_out=1, loss_func=torch.nn.SmoothL1Loss())
24
- learner.fit_one_cycle(1, 1e-3)
25
 
26
 
27
  if __name__ == "__main__":
 
13
 
14
  def create_data(data_path):
15
  fnames = get_files(data_path/'train', extensions='.jpg')
16
+ data = SegmentationDataLoaders.from_label_func(data_path/'train', bs=1, num_workers=0, fnames=fnames, label_func=get_y_fn)
17
  return data
18
 
19
 
20
  def train(data):
 
 
21
  learner = unet_learner(data, resnet34, metrics=rmse, wd=1e-2, n_out=1, loss_func=torch.nn.SmoothL1Loss())
22
+ learner.fine_tune(8)
23
 
24
 
25
  if __name__ == "__main__":