myasin commited on
Commit
3e88d4b
·
1 Parent(s): 3788e82

Update train_net.py

Browse files
Files changed (1) hide show
  1. 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