-
Notifications
You must be signed in to change notification settings - Fork 143
Conversation
/assign @richardsliu Can you create an image and provide a tag? I will update then Related: kubeflow/katib#283 |
from tensorboardX import SummaryWriter | ||
writer = SummaryWriter('runs') | ||
# Training settings | ||
parser = argparse.ArgumentParser(description='PyTorch MNIST Example') |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Do you need to parameterize the output directory? Where do the events get stored?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
updated
/lgtm |
[APPROVALNOTIFIER] This PR is APPROVED This pull-request has been approved by: richardsliu The full list of commands accepted by this bot can be found here. The pull request process is described here
Needs approval from an approver in each of these files:
Approvers can indicate their approval by writing |
Example of Pytorch mnist job in Katib
This change is