未验证 提交 8d4ccb19 编写于 作者: Q qingqing01 提交者: GitHub

Fix args. (#1298)

上级 19b3be8f
......@@ -126,7 +126,7 @@ def train(args,
devices = os.getenv("CUDA_VISIBLE_DEVICES") or ""
devices_num = len(devices.split(","))
batch_size = train_params['batch_size']
epoc_num = train_params['epoch_num']
epoc_num = train_params['epoc_num']
batch_size_per_device = batch_size // devices_num
iters_per_epoc = train_params["train_images"] // batch_size
num_workers = 8
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册