Internal change
PiperOrigin-RevId: 295821819 Change-Id: I7307e94062e3020ec26896634cfc23041773ff8e
This commit is contained in:
parent
79fdda5ead
commit
b391cb55c2
|
@ -124,6 +124,8 @@ class Conv(Layer):
|
||||||
activity_regularizer=regularizers.get(activity_regularizer),
|
activity_regularizer=regularizers.get(activity_regularizer),
|
||||||
**kwargs)
|
**kwargs)
|
||||||
self.rank = rank
|
self.rank = rank
|
||||||
|
if filters is not None and not isinstance(filters, int):
|
||||||
|
filters = int(filters)
|
||||||
self.filters = filters
|
self.filters = filters
|
||||||
self.kernel_size = conv_utils.normalize_tuple(
|
self.kernel_size = conv_utils.normalize_tuple(
|
||||||
kernel_size, rank, 'kernel_size')
|
kernel_size, rank, 'kernel_size')
|
||||||
|
|
Loading…
Reference in New Issue