-
Notifications
You must be signed in to change notification settings - Fork 1
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Adam tv detection #16
base: main
Are you sure you want to change the base?
Conversation
Please remove the WIP from the readme next to the seg and detect model listings |
I ran both of the detection models overnight and they don't appear to be making progress every epoch. Can we please ensure that if progress IS being made, it is being printed out? And if progress is NOT being made, please instrument the startup with printouts of the various steps it needs to go through before making progress and how long they take. e.g. imports, process group setup, model construction, dataset construction, dataloader construction, loading from checkpoint, ddp setup etc. |
I will review and run the segmentation fixes today too. |
…ated timer function
…0 for legible reporting on imports
Hi @adam-peaston-SC can you do a general clean up:
|
Please delete the logs file. |
How's this going? |
tv-detection & tv-segmentation
Description
Commit includes model training scripts for the "tv-detection" and "tv-segmentation" validated configurations sections.
Model / backbone details
MaskRCNN: Resnet101
RetinaNet: Resnet101
Deeplabv3: Mobilenet_v3_large
FCN: ResNet101
Training parameters
MaskRCNN: nodes=11, batch_size=2 (eff. 132), lr=0.02
RetinaNet: nodes=11, batch_size=2 (eff. 132), lr=0.01
Deeplapv3: nodes=11, batch_size=2 (eff. 132), lr=0.01
FCN: nodes=11, batch_size=4, (eff. 264) lr=0.02
Results
Gists of training logs:
MaskRCNN: https://gist.github.com/adam-peaston-SC/e5e5f3dbd1469bf8d7bd0e8d41f471ed
RetinaNet: https://gist.github.com/adam-peaston-SC/5d476bf2beafe9724c3cf21f339110bc
Deeplabv3: https://gist.github.com/adam-peaston-SC/8e09722761712c5f9badf68ec4f8831b
FCN: https://gist.github.com/adam-peaston-SC/730da428954ee6b4299c270946f33bdd
Things done