mirror of
https://git.freebsd.org/ports.git
synced 2025-05-14 08:11:50 -04:00
8f28236546
Sponsored by: Future Crew, LLC Pull Request: https://github.com/freebsd/freebsd-ports/pull/361
11 lines
501 B
Python
11 lines
501 B
Python
--- word_language_model/main.py.orig 2025-03-17 17:25:58 UTC
|
|
+++ word_language_model/main.py
|
|
@@ -243,7 +243,7 @@ with open(args.save, 'rb') as f:
|
|
|
|
# Load the best saved model.
|
|
with open(args.save, 'rb') as f:
|
|
- model = torch.load(f)
|
|
+ model = torch.load(f, weights_only=False)
|
|
# after load the rnn params are not a continuous chunk of memory
|
|
# this makes them a continuous chunk, and will speed up forward pass
|
|
# Currently, only rnn model supports flatten_parameters function.
|