sparse_autoencoder/config.toml

13 lines
171 B
TOML
Raw Normal View History

batch_size = 64
2023-12-23 15:32:18 +01:00
steps = 10000
print_every = 500
seed = 1
cnn_storage = "./res/cnn.eqx"
2023-12-13 20:42:30 +01:00
[[sae]]
layer = 6
2023-12-23 15:32:18 +01:00
hidden_size = 256
2023-12-13 20:42:30 +01:00
input_size = 64
2023-12-14 13:44:35 +01:00
learning_rate = 1e-4
l1 = 3e-4