Commit e76ed01

authored

Add missing name attribute to deeplabv3 & deeplabv3plus (#1214)

1 parent 38af4fc commit e76ed01Copy full SHA for e76ed01

File tree

-0

lines changed

-0

lines changed

Lines changed: 2 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -119,6 +119,7 @@ def __init__(`
`119`	`119`	`)`
`120`	`120`	`else:`
`121`	`121`	`self.classification_head = None`
	`122`	`+ self.name = "deeplabv3-{}".format(encoder_name)`
`122`	`123`
`123`	`124`	`def load_state_dict(self, state_dict, args, *kwargs):`
`124`	`125`	`# For backward compatibility, previously Decoder module was Sequential`
`@@ -238,3 +239,4 @@ def __init__(`
`238`	`239`	`)`
`239`	`240`	`else:`
`240`	`241`	`self.classification_head = None`
	`242`	`+ self.name = "deeplabv3plus-{}".format(encoder_name)`

Comments

(0)