updates
This commit is contained in:
parent
1e62ee2152
commit
109991198c
2
train.py
2
train.py
|
@ -128,7 +128,7 @@ def train(
|
||||||
# plt.savefig('LR.png', dpi=300)
|
# plt.savefig('LR.png', dpi=300)
|
||||||
|
|
||||||
# Dataset
|
# Dataset
|
||||||
rectangular_training = True
|
rectangular_training = False
|
||||||
dataset = LoadImagesAndLabels(train_path,
|
dataset = LoadImagesAndLabels(train_path,
|
||||||
img_size,
|
img_size,
|
||||||
batch_size,
|
batch_size,
|
||||||
|
|
Loading…
Reference in New Issue