Need to disable default logging

[calliope-pro]

Running a simple MambaBlock model automatically creates a logs directory and generates log files, even with minimal code:

class Model(nn.Module):
    def __init__(self, d_model: int, output_size: int):
        super().__init__()
        self.block = MambaBlock(
            dim=d_model,
            depth=1,
            d_conv=16,
        )
        self.fc = nn.Linear(d_model, output_size)
    def forward(self, x):
        x = self.block(x)
        x = self.fc(x)
        return x
if __name__ == "__main__":
    D_MODEL = 64
    OUTPUT_SIZE = 10
    x = torch.randn(1, 10, D_MODEL, device="cuda")
    model = Model(D_MODEL, OUTPUT_SIZE).to("cuda")
    y = model(x)
    print(f"{x.shape=}, {y.shape=}")

Please suggest ways to prevent this automatic logging.

(This English text was written with AI assistance)