added workaround for dotnet/TorchSharp#372

(save is moving model to CPU)
losttech · Nov 17, 2021 · d961e58 · d961e58
1 parent 4df5623
commit d961e58
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/asiren/CompressCommand.cs b/asiren/CompressCommand.cs
@@ -1,4 +1,4 @@
-namespace LostTech.Torch.NN {
+namespace LostTech.Torch.NN {
     using System;
     using System.Diagnostics;
     using System.IO;
@@ -70,6 +70,9 @@ public override int Run(string[] remainingArguments) {
                 Console.WriteLine($"saved! epoch: {eventArgs.Epoch} loss: {eventArgs.AvgLoss}");
 
                 lastUpgrade = eventArgs.Epoch;
+
+                // workaround for https://github.com/dotnet/TorchSharp/issues/372
+                siren = siren.to(device);
             });
 
             int batchesPerEpoch = samples.Length / this.BatchSize;