未验证 提交 ba1a78d8 编写于 作者: B blue-fish 提交者: GitHub

Remove 2 lines not needed for CPU training (#391)

上级 1e168774
......@@ -10,8 +10,6 @@ def sync(device: torch.device):
# For correct profiling (cuda operations are async)
if device.type == "cuda":
torch.cuda.synchronize(device)
else:
torch.cpu.synchronize(device)
def train(run_id: str, clean_data_root: Path, models_dir: Path, umap_every: int, save_every: int,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册