cellpose-web/cp_train.py