mxnet-finetuner

mxnet-finetuner copied to clipboard

Hello,

I really enjoy this! I hope that adding the following features would be more helpful. 1: The name of the training data directory other than /image can be selected by config.yml. Otherwise, I kept changing the folder name to "image". 2: plotting training/validation losses 3: The model folder is full of saved checkpoints for each epoch (already 400 GB in my case, too painful with someone using a compact PC setting). config.yml to specify how to keep the checkpoint data (save all, save the best one only, save the last one only, etc.).

Thanks!!

Thanks for suggesting features. And sorry for my late reply.

Training/validation loss plotting function is currently being implemented.

I plan to implement the functions of No. 1 and 3 as well, but I do not know when that will be.