search

princeton-vl / CornerNet-Lite

BSD 3-Clause "New" or "Revised" License
1.78k stars 431 forks source link

How Can I Use a Pre_trained Model? #128

Open DataXujing opened 4 years ago

DataXujing commented 4 years ago

How can I use a pretraied model to train cornernet-lite?

qrt159 commented 4 years ago

usage: train.py [-h] [--iter START_ITER] [--workers WORKERS] [--initialize] [--distributed] [--world-size WORLD_SIZE] [--rank RANK] [--dist-url DIST_URL] [--dist-backend DIST_BACKEND] cfg_file eg：python train.py CornerNet_Saccade --iter 6000

DataXujing commented 4 years ago

and when training my own dataset , can I use transfer learning，because, my dataset is small.

mad-fogs commented 4 years ago

Hello, how to load the coco pretrained model for transfer learning on custom datasets? any tip here?

thank very much.