Note: Some files are omitted in the project tree above for the sake of brevity. Note that batch size is 16 by default. You may change it to speed up the training process or to avoid memory issues. You ...