This commit is contained in:
Starbeamrainbowlabs 2023-02-02 15:48:49 +00:00
parent f7666865a0
commit 71088b8c0b
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -105,7 +105,7 @@ def make_encoderonly(windowsize, channels, encoder="convnext", water_bins=2, ste
model = make_encoderonly(
windowsize=WINDOW_SIZE,
channels=CHANNELS.
channels=CHANNELS,
steps_per_execution=STEPS_PER_EXECUTION
)
summarywriter(model, os.path.join(DIRPATH_OUTPUT, "summary.txt"))