Created
June 3, 2020 15:22
-
-
Save arunm8489/27bbc18b1939937b266d0f9a659069bd to your computer and use it in GitHub Desktop.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| def forward(self, x, CUDA=False): | |
| modules = self.blocks[1:] | |
| #We cache the outputs for the route layer | |
| outputs = {} | |
| write = 0 | |
| for i, module in enumerate(modules): | |
| module_type = (module["type"]) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment