Namespace(batch=128, cuda=True, lr=0.0001, mode='train', model_name='backup-model.pth', num_classes=2, num_epochs=11, prediction_file='prediction.txt', print_iter=10)
Darknet53(
(conv1): Sequential(
(0): Conv2d(3, 32, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
(1): BatchNorm2d(32, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
(2): LeakyReLU(negative_slope=0.01)
)