mirror of
https://github.com/gryf/coach.git
synced 2025-12-18 11:40:18 +01:00
Add Flatten layer to architectures + make flatten optional in embedders (#483)
Flatten layer required for embedders that mix conv and dense (Cherry picking from #478)
This commit is contained in:
@@ -76,3 +76,11 @@ class NoisyNetDense(object):
|
||||
|
||||
def __str__(self):
|
||||
return "Noisy Dense (num outputs = {})".format(self.units)
|
||||
|
||||
|
||||
class Flatten(object):
|
||||
"""
|
||||
Base class for framework specific flatten layer (used to convert 3D convolution output to 1D dense input)
|
||||
"""
|
||||
def __str__(self):
|
||||
return "Flatten"
|
||||
|
||||
Reference in New Issue
Block a user