20 lines
202 B
Python
20 lines
202 B
Python
from enum import Enum
|
|
from jax import jit
|
|
|
|
class NetworkType(Enum):
|
|
ANN = 0
|
|
SNN = 1
|
|
LSTM = 2
|
|
|
|
|
|
|
|
|
|
@jit
|
|
def func(d):
|
|
return d[0] + 1
|
|
|
|
|
|
d = {0: 1, 1: NetworkType.ANN.value}
|
|
|
|
print(func(d))
|