reset shared.state.is_hr_pass on begin but not end
This commit is contained in:
parent
1d365d2a93
commit
3db29ed99c
|
@ -223,7 +223,6 @@ class State:
|
|||
def end(self):
|
||||
self.job = ""
|
||||
self.job_count = 0
|
||||
self.is_hr_pass = False
|
||||
|
||||
devices.torch_gc()
|
||||
|
||||
|
|
Loading…
Reference in New Issue
Block a user