diff options
author | Jean-Marc Valin <jmvalin@jmvalin.ca> | 2024-01-17 10:26:48 +0300 |
---|---|---|
committer | Jean-Marc Valin <jmvalin@jmvalin.ca> | 2024-01-17 10:26:48 +0300 |
commit | 4f311a1ad44f1b7bd60e32984ca0604c46b6c593 (patch) | |
tree | 7bc6041a00e98dd1ff926253e68cffb2c32ece6f /dnn/torch/plc/train_plc.py | |
parent | 26ddfd713537accce773acc12f565021f4f6d28c (diff) |
PLC export script
mostly untested
Diffstat (limited to 'dnn/torch/plc/train_plc.py')
-rw-r--r-- | dnn/torch/plc/train_plc.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/dnn/torch/plc/train_plc.py b/dnn/torch/plc/train_plc.py index 97be2c04..12b31c4e 100644 --- a/dnn/torch/plc/train_plc.py +++ b/dnn/torch/plc/train_plc.py @@ -138,7 +138,7 @@ if __name__ == '__main__': ) # save checkpoint - checkpoint_path = os.path.join(checkpoint_dir, f'fargan{args.suffix}_{epoch}.pth') + checkpoint_path = os.path.join(checkpoint_dir, f'plc{args.suffix}_{epoch}.pth') checkpoint['state_dict'] = model.state_dict() checkpoint['loss'] = running_loss / len(dataloader) checkpoint['epoch'] = epoch |