Skip to content
Snippets Groups Projects
Commit 20e7b9e9 authored by Alexandre Chapin's avatar Alexandre Chapin :race_car:
Browse files

Fix config issue

parent 4c0c4e81
No related branches found
No related tags found
No related merge requests found
......@@ -97,7 +97,7 @@ def main():
shuffle=True, worker_init_fn=data.worker_init_fn)
#### Create model
model = SlotAttentionAutoEncoder(resolution, num_slots, num_iterations).to(device)
model = SlotAttentionAutoEncoder(resolution, num_slots, num_iterations, cfg=cfg).to(device)
num_params = sum(p.numel() for p in model.parameters())
print('Number of parameters:')
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment