#!/usr/bin/env bash
# Usage:
# ./train.sh \

mkdir logs/train

export PYTHONUNBUFFERED="True"

GPU_ID=0

LOG="logs/train/VGG16.txt.`date +'%Y-%m-%d_%H-%M-%S'`"
exec &> >(tee -a "$LOG")
echo Logging output to "$LOG"


# CUDA_VISIBLE_DEVICES=${GPU_ID}
# time ./src/train_net.py --gpu ${GPU_ID} \
#   --solver model/prototxt/solver.prototxt \
#   --weights model/imagenet_models/VGG16.v2.caffemodel \
#   --imdb icdar_2015_train \
#   --iters 200000 \
#   --cfg src/cfgs/faster_rcnn_end2end_ohem.yml \

  # --weights logs/train/snapshots_step_bn/_iter_200000.caffemodel\
  # --weights model/imagenet_models/VGG16.v2.caffemodel\
  # --iters 80000 \

time ./src/train_net.py --gpu ${GPU_ID} \
  --solver model/prototxt/solver_step_bn.prototxt \
  --weights model/imagenet_models/VGG16.v2.caffemodel\
  --imdb icdar_2015_train \
  --iters 200000 \
  --cfg src/cfgs/faster_rcnn_end2end_ohem.yml \
  --snapshots_dir logs/train/snapshots_step_bn/

  # --imdb ship_2021_train \
# --imdb icdar_2015_train \

# R2CNN/logs/train/snapshots/_iter_70000.caffemodel