GBML
GBML copied to clipboard
关于支持集和查询及
trafficstars
reptile不需要划分支持集和查询及,但是你这里读取了支持集和查询及,此外为什么对于每个任务训练的时候 ,又从支持集读取了10张图片呢?这到底是5way 几shot呢? override = self.inner_optimizer if is_train else None #其实是相当于创建了一个副本,这个优化器和网络的更新都不会影响原来的 with higher.innerloop_ctx(self.network, self.inner_optimizer, track_higher_grads=False) as (fmodel, diffopt):
for step in range(self.args.n_inner):
if is_train:
index = np.random.permutation(np.arange(len(test_input)))[:10]
train_input = test_input[index]
train_target = test_target[index]
self.inner_loop(fmodel, diffopt, train_input, train_target)