提交 c5a29dba 编写于 作者: Z zh-hike 提交者: Walter

代码规范

上级 f073e97d
......@@ -170,4 +170,4 @@ class AttentionModel(DistillationModel):
else:
out = self.model_list[idx](out, label)
result_dict.update(out)
return result_dict
return result_dict
\ No newline at end of file
......@@ -202,7 +202,7 @@ class Wide_ResNet(TheseusLayer):
feat = self.relu(self.bn1(feat))
feat = F.adaptive_avg_pool2d(feat, 1)
feat = paddle.reshape(feat, [-1, self.channels])
if self.proj:
pfeat = self.fc1(feat)
pfeat = self.relu_mlp(pfeat)
......
......@@ -103,7 +103,7 @@ DataLoader:
order: hwc
sampler:
name: DistributedBatchSampler # DistributedBatchSampler
name: DistributedBatchSampler
batch_size: 64
drop_last: true
shuffle: true
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册