WANDB_MODE=disabled LOGURU_LEVEL=INFO OMP_NUM_THREADS=4 CUDA_VISIBLE_DEVICES=0,1 torchrun --nnodes=1 --nproc_per_node=2 --master_port=7888 train.py --config_file Merfishbrain-train.yaml --data_path ./data_merfish.yaml
