Skip to content

Commit

Permalink
No public description
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 561465374
  • Loading branch information
yilei authored and tensorflower-gardener committed Aug 30, 2023
1 parent 2c14ca3 commit ef1471a
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion official/projects/centernet/configs/centernet.py
Original file line number Diff line number Diff line change
Expand Up @@ -71,7 +71,7 @@ class DataConfig(cfg.DataConfig):
is_training: bool = True
dtype: str = 'float16'
decoder: DataDecoder = dataclasses.field(default_factory=DataDecoder)
parser: Parser = Parser()
parser: Parser = dataclasses.field(default_factory=Parser)
shuffle_buffer_size: int = 10000
file_type: str = 'tfrecord'
drop_remainder: bool = True
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@ class ClassificationExampleConfig(cfg.TaskConfig):
# At most one of `init_checkpoint` and `hub_module_url` can be specified.
init_checkpoint: str = ''
hub_module_url: str = ''
model: ModelConfig = ModelConfig()
model: ModelConfig = dataclasses.field(default_factory=ModelConfig)

num_classes = 2
class_names = ['A', 'B']
Expand Down

0 comments on commit ef1471a

Please sign in to comment.