Customise order of tails
Right now the main tail has to be the last tail listed in loss etc.
Instead, we should use a flag tail_format='main_last'
, like keras is using data_format
for channels first/ last for conv layers.
The default in mlt should be 'main_last'
Edited by Ghost User