MCPcopy Create free account
hub / github.com/DSL-Lab/GRBM / load

Function load

main.py:26–28  ·  view source on GitHub ↗
(model, results_folder, epoch)

Source from the content-addressed store, hash-verified

24
25
26def load(model, results_folder, epoch):
27 data = torch.load(f'{results_folder}/model-{epoch}.pt')
28 model.load_state_dict(data['model'])
29
30
31def train(model,

Callers 1

train_modelFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected