Skip to content

Commit

Permalink
update 1 file and copy 1 file: add aux training scripts
Browse files Browse the repository at this point in the history
  • Loading branch information
dummyindex committed Mar 18, 2024
1 parent 3e6960e commit 6b88cf9
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -5,5 +5,5 @@
# model 901
# model 88 resume
# model 75 aug
model=v11_01_aux
model=v11_01_aux-test
nohup python ../livecellx/model_zoo/segmentation/train_csn.py --train_dir="./notebook_results/a549_ccp_vim/train_data_v11" --test_dir="./notebook_results/a549_ccp_vim/test_data_v11" --source=all --model_version=version_$model --epochs=5000 --kernel_size=1 --batch_size=2 --batch_size=2 --degrees=30 --translation=0.3 --aug_scale="0.5,2" --input_type=raw_aug_duplicate --loss=BCE --ou_aux> train_out_model"$model"_resume.out 2>&1 &
9 changes: 9 additions & 0 deletions notebooks/scripts/csn_scripts/train_data_real_v12_01_aux.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
############################################################################################################
# syn underseg training 5300 series) with mild augmentation
############################################################################################################

# model 901
# model 88 resume
# model 75 aug
model=v12_01_aux
nohup python ../livecellx/model_zoo/segmentation/train_csn.py --train_dir="./notebook_results/a549_ccp_vim/train_data_v12" --test_dir="./notebook_results/a549_ccp_vim/test_data_v12" --source=all --model_version=version_$model --epochs=5000 --kernel_size=1 --batch_size=2 --batch_size=6 --degrees=30 --translation=0.3 --aug_scale="0.5,2" --input_type=raw_aug_duplicate --loss=BCE --ou_aux> train_out_model"$model"_resume.out 2>&1 &

0 comments on commit 6b88cf9

Please sign in to comment.