Whatever

ConnorJL · Nov 19, 2017 · c99c740 · c99c740
1 parent e5c694b
commit c99c740
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/train.py b/train.py
@@ -51,6 +51,8 @@ def save_checkpoint(state, filename='checkpoint'):
  print("=> loaded checkpoint '{}' (epoch {})".format(resume_path, checkpoint['epoch']))
  model.cuda()
 
+ print(model.primary_capsules.shape())
+
  for i in range(start_epoch, num_epochs):
  for e in train_dataloader:
  model.zero_grad()