Currently, we really only need an embedding_net in the load_nde_sbi function, and passing it to the Trainer is only necessary if you want to use the data to initialize an architecture.
This can be made simpler by removing the Trainer's init requirement of embedding_net, and instead calling the net initializers directly from the nets objects.
Currently, we really only need an
embedding_netin theload_nde_sbifunction, and passing it to the Trainer is only necessary if you want to use the data to initialize an architecture.This can be made simpler by removing the Trainer's init requirement of embedding_net, and instead calling the net initializers directly from the
netsobjects.