Function save_checkpoint

train.py:197–199 · view source on GitHub ↗

(state, file_prefix)

Source from the content-addressed store, hash-verified

195
196
197	def save_checkpoint(state, file_prefix):
198	file_name = file_prefix + "_" + str(state["epoch"])
199	torch.save(state, file_name)
200
201
202	def train(conf):

trainFunction · 0.85

saveMethod · 0.80

no test coverage detected