Import argparse and sys python modules

eic · Jan 12, 2024 · 92bc46e · 92bc46e
1 parent 45ffea0
commit 92bc46e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/benchmarks/roman_pots/train_dense_neural_network.py b/benchmarks/roman_pots/train_dense_neural_network.py
@@ -6,6 +6,8 @@
 import torch.optim as optim
 import torch.optim.lr_scheduler as lr_scheduler
 import matplotlib.pyplot as plt
+import argparse
+import sys
 torch.set_default_dtype(torch.float32)
 
 if torch.cuda.is_available():