5,350
edits
Line 3: | Line 3: | ||
If you're using a custom training loop (i.e. gradient tape), then you'll need to set everything up manually. | If you're using a custom training loop (i.e. gradient tape), then you'll need to set everything up manually. | ||
First create a <code>SummaryWriter<code> | First create a <code>SummaryWriter</code> | ||
<syntaxhighlight lang="python"> | <syntaxhighlight lang="python"> | ||
train_log_dir = os.path.join(args.checkpoint_dir, "logs", "train") | train_log_dir = os.path.join(args.checkpoint_dir, "logs", "train") |