Spaces:

aim4composites
/

Additive_Manufacturing_Design

Sleeping

ryanone commited on 14 days ago

Commit

bfbe97d

verified ·

1 Parent(s): 459841c

Update model_fdm.py

Files changed (1) hide show

model_fdm.py CHANGED Viewed

@@ -4,7 +4,7 @@ import matplotlib.pyplot as plt
 from Dataset import DataAdditiveManufacturing, DataThermoforming
 from model import NeuralNetwork
-DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 # Set global plotting parameters
 plt.rcParams.update({'font.size': 14,
                      'figure.figsize': (10, 8),

 from Dataset import DataAdditiveManufacturing, DataThermoforming
 from model import NeuralNetwork
+DEVICE = torch.device('cpu')
 # Set global plotting parameters
 plt.rcParams.update({'font.size': 14,
                      'figure.figsize': (10, 8),