myasin
commited on
Commit
·
3e88d4b
1
Parent(s):
3788e82
Update train_net.py
Browse files- tools/train_net.py +3 -0
tools/train_net.py
CHANGED
@@ -56,6 +56,9 @@ class Trainer(DefaultTrainer):
|
|
56 |
are working on a new research project. In that case you can use the cleaner
|
57 |
"SimpleTrainer", or write your own training loop. You can use
|
58 |
"tools/plain_train_net.py" as an example.
|
|
|
|
|
|
|
59 |
"""
|
60 |
|
61 |
@classmethod
|
|
|
56 |
are working on a new research project. In that case you can use the cleaner
|
57 |
"SimpleTrainer", or write your own training loop. You can use
|
58 |
"tools/plain_train_net.py" as an example.
|
59 |
+
|
60 |
+
Adapted from:
|
61 |
+
https://github.com/facebookresearch/detectron2/blob/master/projects/DeepLab/train_net.py
|
62 |
"""
|
63 |
|
64 |
@classmethod
|