forked from zzzxxxttt/pytorch_simple_CenterNet_45
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtrain.sh
3 lines (3 loc) · 1.33 KB
/
train.sh
1
2
3
# CUDA_VISIBLE_DEVICES=0 nvprof --profile-from-start off -o trace_name.prof python profiler.py --data_dir=/tmp2/igor/EV/Dataset/Automotive/ --log_name=test --train_csv_file=/tmp2/igor/EV/EventCenterNet/tests/train_a.csv --val_csv_file=/tmp2/igor/EV/EventCenterNet/tests/val_a.csv --class_list_file=/tmp2/igor/EV/EventCenterNet/datasets/classes.csv --trim_to_shortest --delta_t=50000 --frames_per_batch=5 --bins=5 --batch_size=2 --val_interval=1 --log_interval=1 --num_epochs=1
# CUDA_VISIBLE_DEVICES=0 python train.py --data_dir=/tmp2/igor/EV/Dataset/Automotive/ --log_name=test --train_csv_file=/tmp2/igor/EV/EventCenterNet/tests/train_a.csv --val_csv_file=/tmp2/igor/EV/EventCenterNet/tests/val_a.csv --class_list_file=/tmp2/igor/EV/EventCenterNet/datasets/classes.csv --trim_to_shortest --delta_t=50000 --frames_per_batch=10 --bins=5 --batch_size=2 --val_interval=1 --log_interval=1 --num_epochs=10 --lr=0.000005
CUDA_VISIBLE_DEVICES=1 python train.py --data_dir=/tmp2/igor/EV/Dataset/Automotive/ --log_name=test_debug --train_csv_file=/tmp2/igor/EV/EventCenterNet/datasets/train_a.csv --val_csv_file=/tmp2/igor/EV/EventCenterNet/datasets/val_a.csv --class_list_file=/tmp2/igor/EV/EventCenterNet/datasets/classes.csv --trim_to_shortest --delta_t=50000 --frames_per_batch=5 --bins=5 --batch_size=3 --val_interval=1 --log_interval=1 --num_epochs=10 --lr=0.0005