Commit 0126950d authored by Vishnu Banna's avatar Vishnu Banna
Browse files

docstring update

parent 0d3ec509
...@@ -207,7 +207,7 @@ class YoloLossBase(object, metaclass=abc.ABCMeta): ...@@ -207,7 +207,7 @@ class YoloLossBase(object, metaclass=abc.ABCMeta):
@abc.abstractmethod @abc.abstractmethod
def _build_per_path_attributes(self): def _build_per_path_attributes(self):
"""Additional initialization required for each YOLO loss version""" """Additional initialization required for each YOLO loss version."""
... ...
@abc.abstractmethod @abc.abstractmethod
...@@ -401,7 +401,7 @@ class DarknetLoss(YoloLossBase): ...@@ -401,7 +401,7 @@ class DarknetLoss(YoloLossBase):
ind_mask, grid_mask) ind_mask, grid_mask)
def cross_replica_aggregation(self, loss, num_replicas_in_sync): def cross_replica_aggregation(self, loss, num_replicas_in_sync):
"""this method is not specific to each loss path, but each loss type""" """This method is not specific to each loss path, but each loss type."""
return loss / num_replicas_in_sync return loss / num_replicas_in_sync
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment