skypilot

1
name: detectron2-app
2

3
resources:
4
  accelerators: V100
5

6
setup: |
7
  pip install --upgrade pip
8

9
  conda activate d2
10

11
  if [ $? -eq 0 ]; then
12
    echo "conda env exists"
13
  else
14
    conda create -n d2 python=3.8 -y
15
    conda activate d2
16
    pip install torch torchvision
17
    git clone https://github.com/facebookresearch/detectron2
18
    cd detectron2
19
    pip install -e .
20
  fi
21

22
  cd tools
23
  mkdir -p datasets/coco
24
  cd datasets/coco
25

26
  wget http://images.cocodataset.org/zips/train2017.zip
27
  wget http://images.cocodataset.org/zips/val2017.zip
28

29
  unzip train2017.zip
30
  unzip val2017.zip
31

32
  rm train2017.zip
33
  rm val2017.zip
34

35
  wget http://images.cocodataset.org/annotations/annotations_trainval2017.zip
36
  wget http://images.cocodataset.org/annotations/stuff_annotations_trainval2017.zip
37

38
  unzip annotations_trainval2017.zip
39
  unzip stuff_annotations_trainval2017.zip
40

41
  rm annotations_trainval2017.zip
42
  rm stuff_annotations_trainval2017.zip
43

44
run: |
45
  conda activate d2
46
  
47
  cd detectron2/tools
48
  python train_net.py --config-file ../configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x.yaml --num-gpus 1 SOLVER.IMS_PER_BATCH 2 SOLVER.BASE_LR 0.0025 SOLVER.MAX_ITER 100
49