| 73 | return x |
| 74 | |
| 75 | def pre_encode(self,x): |
| 76 | |
| 77 | if self.args.dataset in ['cifar100','manyshotcifar']: |
| 78 | x = self.encoder.conv1(x) |
| 79 | x = self.encoder.bn1(x) |
| 80 | x = self.encoder.relu(x) |
| 81 | x = self.encoder.layer1(x) |
| 82 | x = self.encoder.layer2(x) |
| 83 | |
| 84 | elif self.args.dataset in ['mini_imagenet','manyshotmini','cub200']: |
| 85 | x = self.encoder.conv1(x) |
| 86 | x = self.encoder.bn1(x) |
| 87 | x = self.encoder.relu(x) |
| 88 | x = self.encoder.maxpool(x) |
| 89 | x = self.encoder.layer1(x) |
| 90 | x = self.encoder.layer2(x) |
| 91 | |
| 92 | return x |
| 93 | |
| 94 | |
| 95 | def post_encode(self,x): |