Research Article
A Novel Memory-Scheduling Strategy for Large Convolutional Neural Network on Memory-Limited Devices
Algorithm 1
The Memory-Management Strategy of Control Module in mmCNN.
| Input: | | platform configuration | | network model | | up bound of partition number | | Output: | | memory schedule list | | the divided number of prefetched convolutional layer | | the divided number of this layer | (1) | Calculate the number of , | (2) | for do | (3) | From and calculate and part number in current | (4) | | (5) | if then | (6) | Calculate | (7) | else | (8) | | (9) | Calculate ’s self number | (10) | end if | (11) | Reset | (12) | end for | (13) | Return |
|