(self, x)
| 903 | nn.init.constant_(m.bias, 0) |
| 904 | |
| 905 | def forward(self, x): |
| 906 | fake_prob_list = [] |
| 907 | meta_embedding = torch.transpose(x, 1, 2) |
| 908 | for att_head in self.attribute_generator_list: |
| 909 | fake_prob = att_head(meta_embedding) |
| 910 | fake_prob_list.append(fake_prob) |
| 911 | return fake_prob_list |
| 912 | |
| 913 | |
| 914 | def at_small(pretrain_path): |
nothing calls this directly
no outgoing calls
no test coverage detected