dbartlett/mgborg_emulator
0
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add mini model weights

Browse source
This commit is contained in:
Deaglan Bartlett 2024-06-14 16:34:27 +02:00
parent 2016ef0599
commit b966c4f95e
6 changed files with 97 additions and 93 deletions
Download patch file Download diff file Expand all files Collapse all files

10
conf/basic_ini.ini
View file

@ -4,9 +4,9 @@ VERBOSE_LEVEL = 2
N0 = 128
N1 = 128
N2 = 128
L0 = 500.0
L1 = 500.0
L2 = 500.0
L0 = 250.0
L1 = 250.0
L2 = 250.0
corner0 = -250.0
corner1 = -250.0
corner2 = -250.0
@ -66,8 +66,8 @@ z0 = 0
[emulator]
use_emulator = True
model_weights_path = /home/bartlett/mgborg_emulator/weights/d2d_borg.pt
architecture = StyledVNet
model_weights_path = /home/bartlett/mgborg_emulator/weights/d2d_borg_mini.pt
architecture = StyledVNet_distill
use_float64 = False
use_pad_and_NN = True
requires_grad = False