foscat.loss_backend_torch#

Classes#

Module Contents#

class foscat.loss_backend_torch.loss_backend(backend, curr_gpu, mpi_rank)[source]#

bk#

curr_gpu#

mpi_rank#

check_dense(data, datasz)[source]#

loss(x, batch, loss_function, KEEP_TRACK)[source]#