Update model_fdm.py
Browse files- model_fdm.py +1 -0
model_fdm.py
CHANGED
|
@@ -5,6 +5,7 @@ from Dataset import DataAdditiveManufacturing, DataThermoforming
|
|
| 5 |
from model import NeuralNetwork
|
| 6 |
|
| 7 |
DEVICE = torch.device('cpu')
|
|
|
|
| 8 |
# Set global plotting parameters
|
| 9 |
plt.rcParams.update({'font.size': 14,
|
| 10 |
'figure.figsize': (10, 8),
|
|
|
|
| 5 |
from model import NeuralNetwork
|
| 6 |
|
| 7 |
DEVICE = torch.device('cpu')
|
| 8 |
+
|
| 9 |
# Set global plotting parameters
|
| 10 |
plt.rcParams.update({'font.size': 14,
|
| 11 |
'figure.figsize': (10, 8),
|