HoneyTian commited on
Commit
f510138
·
1 Parent(s): 50cfa5e

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=args.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=args.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 系统中不能.