Skip to content

GitLab

Commit f52c8fd5 authored by Yanghan Wang's avatar Yanghan Wang Committed by Facebook GitHub Bot
Browse files

mark all subclass of DefaultTask 

Summary: Pull Request resolved: https://github.com/facebookresearch/d2go/pull/303

Reviewed By: mcimpoi

Differential Revision: D37220449

fbshipit-source-id: 04985a4b0129ef37aa42534cdb5f3f8d3f6abf00
parent dc477dd9
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
d2go/runner/lightning_task.py
View file @ f52c8fd5
...@@ -490,6 +490,7 @@ class DefaultTask(pl.LightningModule): ...@@ -490,6 +490,7 @@ class DefaultTask(pl.LightningModule):
return self return self
# TODO(T123654122): subclass of DefaultTask will be refactored
class GeneralizedRCNNTask(DefaultTask): class GeneralizedRCNNTask(DefaultTask):
@classmethod @classmethod
def get_default_cfg(cls): def get_default_cfg(cls):
......
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