Restore cleanup_beam method (#6446)

This commit is contained in:
Adriane Boyd 2020-11-25 13:21:48 +01:00 committed by GitHub
parent fe9009911a
commit e0f5646a4a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -304,3 +304,7 @@ def get_gradient(nr_class, beam_maps, histories, losses):
grads[j][i, clas] += loss
key = key + tuple([clas])
return grads
def cleanup_beam(Beam beam):
del beam