Spaces:
Paused
Paused
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- 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.
|
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__":
|