Skip to content

Commit 9f1a812

Browse files
committed
Fix SDNE custom History callback in manual training loop
1 parent 56e7bdf commit 9f1a812

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

ge/models/sdne.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -123,6 +123,7 @@ def train(self, batch_size=1024, epochs=1, initial_epoch=0, verbose=1):
123123
else:
124124
steps_per_epoch = (self.node_size - 1) // batch_size + 1
125125
hist = History()
126+
hist.set_model(self.model)
126127
hist.on_train_begin()
127128
logs = {}
128129
for epoch in range(initial_epoch, epochs):

0 commit comments

Comments
 (0)