Spaces:
Running
Running
add frcrn model
Browse files
examples/frcrn/step_2_train_model.py
CHANGED
@@ -135,7 +135,7 @@ def main():
|
|
135 |
)
|
136 |
train_data_loader = DataLoader(
|
137 |
dataset=train_dataset,
|
138 |
-
batch_size=
|
139 |
# shuffle=True,
|
140 |
sampler=None,
|
141 |
# Linux 系统中可以使用多个子进程加载数据, 而在 Windows 系统中不能.
|
@@ -146,7 +146,7 @@ def main():
|
|
146 |
)
|
147 |
valid_data_loader = DataLoader(
|
148 |
dataset=valid_dataset,
|
149 |
-
batch_size=
|
150 |
# shuffle=True,
|
151 |
sampler=None,
|
152 |
# Linux 系统中可以使用多个子进程加载数据, 而在 Windows 系统中不能.
|
|
|
135 |
)
|
136 |
train_data_loader = DataLoader(
|
137 |
dataset=train_dataset,
|
138 |
+
batch_size=config.batch_size,
|
139 |
# shuffle=True,
|
140 |
sampler=None,
|
141 |
# Linux 系统中可以使用多个子进程加载数据, 而在 Windows 系统中不能.
|
|
|
146 |
)
|
147 |
valid_data_loader = DataLoader(
|
148 |
dataset=valid_dataset,
|
149 |
+
batch_size=config.batch_size,
|
150 |
# shuffle=True,
|
151 |
sampler=None,
|
152 |
# Linux 系统中可以使用多个子进程加载数据, 而在 Windows 系统中不能.
|