Round number of filters based on depth multiplier.
(repeats, global_params)
| 72 | |
| 73 | |
| 74 | def round_repeats(repeats, global_params): |
| 75 | """ Round number of filters based on depth multiplier. """ |
| 76 | multiplier = global_params.depth_coefficient |
| 77 | if not multiplier: |
| 78 | return repeats |
| 79 | return int(math.ceil(multiplier * repeats)) |
| 80 | |
| 81 | |
| 82 | def drop_connect(inputs, p, training): |