batchflow
batchflow copied to clipboard
Allow for block chaining in TFModel and TorchModel
All the model block (initial_block, body, head) could be configured as a list of dict. Then each block is added sequentially for each list item.
MobileNet v3 small can benefit from this. So rewrite its config accordingly.
I've done this in #421, and rewritten Xception accordingly. I am not sure that MobileNet would benefit from this, so I did not touch it