sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-12-22 06:05:01 +00:00
Code Issues Projects Releases Packages Wiki Activity

replace tensorflow_addons with keras_cv

Browse source
This commit is contained in:
Starbeamrainbowlabs 2024-08-29 19:32:54 +01:00
parent 2d73d84e20
commit b2b96ab636
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
aimodel/src/lib/ai/components/convnext.py
View file

@ -2,7 +2,7 @@ from unicodedata import name
import numpy as np
import tensorflow as tf
import tensorflow_addons as tfa
import keras_cv
# Code from https://github.com/leanderme/ConvNeXt-Tensorflow/blob/main/ConvNeXt.ipynb
from .LayerConvNeXtGamma import LayerConvNeXtGamma
@ -165,7 +165,8 @@ def add_convnext_block(y, dim, drop_prob=0, prefix=""):
name = f'{prefix}.gamma'
)(y)
y = tfa.layers.StochasticDepth(
# y = tfa.layers.StochasticDepth(
y = keras_cv.layers.StochasticDepth(
drop_prob,
name = f'{prefix}.drop_path'
)([skip, y])