matteo/constellationnet
matteo
/
constellationnet
1
0
Fork 0
Code Issues Pull Requests Releases Activity

Fix variable naming clash

This commit is contained in:
Mattéo Delabre 2019-12-16 10:31:16 -05:00
parent fb2518b321
commit af59421dc0
Signed by: matteo
GPG Key ID: AE3FBD02DC583ABB
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
train.py
View File

@ -47,7 +47,7 @@ running_loss = 0
classes_ordered = torch.arange(order).repeat(batch_size)
# Constellation from the previous training batch
prev_constellation = model.get_constellation()
prev_constel = model.get_constellation()
total_change = float('inf')
# Optimizer settings
@ -79,9 +79,9 @@ while total_change >= 1e-4:
# Check for convergence
model.eval()
constellation = model.get_constellation()
total_change = (constellation - prev_constellation).norm(dim=1).sum()
prev_constellation = constellation
cur_constel = model.get_constellation()
total_change = (cur_constel - prev_constel).norm(dim=1).sum()
prev_constel = cur_constel
# Report loss
running_loss += loss.item()
@ -96,7 +96,7 @@ while total_change >= 1e-4:
# Update figure with current encoding
ax.clear()
util.plot_constellation(
ax, constellation,
ax, cur_constel,
model.channel, model.decoder,
noise_samples=0
)