ITER::1后直接闪退
Closed this issue · 2 comments
kemolo commented
"""
执行到此处开始闪退,f.result()无法执行,没有报错
环境
win10
python 3.7
libtorch 1.3.1
pytorch 1.3.1
"""
#learner.py
with concurrent.futures.ThreadPoolExecutor(max_workers=self.num_train_threads) as executor:
futures = [executor.submit(self.self_play, 1 if itr % 2 else -1, libtorch, k == 1) for k in
range(1, self.num_eps + 1)]
for k, f in enumerate(futures):
examples = f.result()
itr_examples += examples
# decrease libtorch batch size
remain = min(len(futures) - (k + 1), self.num_train_threads)
libtorch.set_batch_size(max(remain * self.num_mcts_threads, 1))
print("EPS: {}, EXAMPLES: {}".format(k + 1, len(examples)))
kemolo commented
已解决
编译之前
libtorch.cpp 第11行
改成 module(std::make_sharedtorch::jit::script::Module(torch::jit::load(model_path.c_str())))
可顺利通过
hijkzzz commented
已经加上显示转换