noble/mmp-osp1
noble/mmp-osp1
Archived
1
0
Fork 0
Code Issues 7 Pull requests Projects 1 Releases Wiki Activity

Added prediction batch size to config

Browse source
This commit is contained in:
Oscar Blue 2022-03-23 10:41:09 +00:00
parent 68ff4ea0ad
commit e29bf123f1
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
src/autophotographer/config.py
View file

@ -14,6 +14,7 @@ PRED_BATCH_SIZE = 4
EPOCHS = 20 EPOCHS = 20
LR = 0.001 LR = 0.001
IMAGE_SIZE = 32 IMAGE_SIZE = 32
PRED_BATCH_SIZE = 4
WARMUP_PLOT = os.path.join("output", "plot.png") WARMUP_PLOT = os.path.join("output", "plot.png")
WARMUP_MODEL = os.path.join("output", "plot.pth") WARMUP_MODEL = os.path.join("output", "plot.pth")