StrongResearch / isc-demos

Deep learning examples for the Instant Super Computer
13 stars 1 forks source link

Detectron2 #25

Open adam-peaston-SC opened 1 year ago

adam-peaston-SC commented 1 year ago

Source/Repository

https://github.com/facebookresearch/detectron2

Model Description

Detectron2

Dataset

COCO

Literature Benchmark Source

https://github.com/facebookresearch/detectron2/blob/main/MODEL_ZOO.md

Literature Benchmark Performance

Image

Strong Compute Results

_Currently cycling on the ISC with the same maximum number of iterations as the literature (270000)...

Configurations (Matching Literature Benchmark Config)

MASK_ON: True RESNETS_DEPTH: 50 BATCH_SIZE_PER_IMAGE: 512 CONFIG_FILE: 'COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml' NUM_WORKERS: 11 CHECKPOINT_PERIOD: 200 IMS_PER_BATCH: 66 BASE_LR: 0.001 MAX_ITER: 270000 STEPS: [210000, 250000]

Logs Gist (Rank 0)

https://gist.github.com/StrongHenrySung/261b25430253c572e4bae49be6fd905a