15 lines
398 B
Python
15 lines
398 B
Python
class SimpleModule:
|
|
|
|
def __init__(self,
|
|
in_features: int,
|
|
out_features: int,
|
|
activation: str = "relu"):
|
|
self.in_features = in_features
|
|
self.out_features = out_features
|
|
self.activation = activation
|
|
print(
|
|
f"SimpleModule created: {in_features} -> {out_features}, activation: {activation}"
|
|
)
|
|
|
|
|