Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
  • Loading branch information
smokhov committed Nov 12, 2021
2 parents 5e7f320 + 2fba707 commit 0d84ba6
Show file tree
Hide file tree
Showing 2 changed files with 41 additions and 1 deletion.
40 changes: 40 additions & 0 deletions src/efficientdet.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,40 @@
#!/encs/bin/tcsh

##
## This script was submitted by a member of Dr. Amer's Research Group
##

##
## Prerequisite:
## Efficientdet Virtual Environment (see commit comments)
##

##
## Job Scheduler options
##

#$ -N efficientdet_pascal
#$ -cwd
#$ -pe smp 8
#$ -l h_vmem=128G
#$ -l gpu=2

cd /speed-scratch/<encs_username>

module load python/3.8.3
module load cuda/11.5
source envs/tf/bin/activate.csh

cd code/automl/efficientdet

python3 main.py --mode=train_and_eval \
--train_file_pattern=tfrecord/'pascal-*-of-00100.tfrecord' \
--val_file_pattern=tfrecord/'val-*-of-00032.tfrecord' \
--model_name='efficientdet-d0' \
--model_dir=model_path \
--backbone_ckpt='efficientnet-b0' \
--train_batch_size=4 \
--eval_batch_size=4 --eval_samples=4952 \
--num_examples_per_epoch=16551 --num_epochs=300 \
--hparams="num_classes=20,moving_average_decay=0,mixed_precision=true" \
--strategy='gpus'
2 changes: 1 addition & 1 deletion src/lambdal-singularity.sh
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
#$ -cwd

# How many GPUs (currently limit is set 2 max for Speed 5 and 17)
#$-l gpu=2
#$ -l gpu=2

# High value of memory requeted
#$ -l h_vmem=20G
Expand Down

0 comments on commit 0d84ba6

Please sign in to comment.