update dllib-mlflow.py
This commit is contained in:
parent
8f297e82de
commit
5f9785d4fd
@ -289,7 +289,7 @@ def my_main(epochs):
|
||||
loss_fn = nn.CrossEntropyLoss()
|
||||
# epochs = 1000
|
||||
# epochs = epochs
|
||||
alpha = int(sys.argv[1]) if len(sys.argv) > 1 else 20
|
||||
epochs = int(sys.argv[1]) if len(sys.argv) > 1 else 20
|
||||
mlflow.log_param("epochs", epochs)
|
||||
|
||||
def print_(loss):
|
||||
|
Loading…
Reference in New Issue
Block a user