Skip to content

Commit

Permalink
Change naming of pc model
Browse files Browse the repository at this point in the history
  • Loading branch information
francesco-innocenti committed Jun 26, 2024
1 parent 012894d commit da313f2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion jpc/_train.py
Original file line number Diff line number Diff line change
Expand Up @@ -100,7 +100,7 @@ def make_pc_step(
mode="unsupervised",
batch_size=batch_size,
sigma=sigma
) if x is None else init_activities_with_ffwd(network=network, x=x)
) if x is None else init_activities_with_ffwd(model=model, x=x)

mse_loss = mean((y - activities[-1])**2) if input is not None else None
equilib_activities = solve_pc_activities(
Expand Down

0 comments on commit da313f2

Please sign in to comment.