- 先設(shè)置不使用梯度咖为,然后將測試時(shí)候的batchsize設(shè)置成訓(xùn)練時(shí)候的二分之一或者三分之一就不會(huì)爆了。
- 可能原因是測試的時(shí)候真的需要更大的顯存徽级。
def early_test(self):
with torch.no_grad():
self.model.eval()
for path in tqdm(self.save_path,desc='Test Loop:'):
self.eval(path)
self.test()