batchflow icon indicating copy to clipboard operation
batchflow copied to clipboard

Allow for block chaining in TFModel and TorchModel

Open roman-kh opened this issue 6 years ago • 2 comments

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.

roman-kh avatar Aug 06 '19 13:08 roman-kh

MobileNet v3 small can benefit from this. So rewrite its config accordingly.

roman-kh avatar Aug 06 '19 13:08 roman-kh

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

SergeyTsimfer avatar Oct 02 '19 08:10 SergeyTsimfer