// File: /workspace/github_c_makefile_star2_O2/SimpleDoger[P]Darknet_CPU_Only/super.c
// Function: train_super
void train_super(char *cfgfile, char *weightfile, int clear)
{
    char *train_images = "/data/imagenet/imagenet1k.train.list";
    char *backup_directory = "/home/pjreddie/backup/";
    srand(time(0));
    char *base = basecfg(cfgfile);
    printf("%s\n", base);
    float avg_loss = -1;
    network *net = load_network(cfgfile, weightfile, clear);
    printf("Learning Rate: %g, Momentum: %g, Decay: %g\n", net->learning_rate, net->momentum, net->decay);
    int imgs = net->batch*net->subdivisions;
    int i = *net->seen/imgs;
    data train, buffer;


    list *plist = get_paths(train_images);
    //int N = plist->size;
    char **paths = (char **)list_to_array(plist);

    load_args args = {0};
    args.w = net->w;
    args.h = net->h;
    args.scale = 4;
    args.paths = paths;
    args.n = imgs;
    args.m = plist->size;
    args.d = &buffer;
    args.type = SUPER_DATA;

    pthread_t load_thread = load_data_in_thread(args);
    clock_t time;
    //while(i*imgs < N*120){
    while(get_current_batch(net) < net->max_batches){
        i += 1;
        time=clock();
        pthread_join(load_thread, 0);
        train = buffer;
        load_thread = load_data_in_thread(args);

        printf("Loaded: %lf seconds\n", sec(clock()-time));

        time=clock();
        float loss = train_network(net, train);
        if (avg_loss < 0) avg_loss = loss;
        avg_loss = avg_loss*.9 + loss*.1;

        printf("%d: %f, %f avg, %f rate, %lf seconds, %d images\n", i, loss, avg_loss, get_current_rate(net), sec(clock()-time), i*imgs);
        if(i%1000==0){
            char buff[256];
            sprintf(buff, "%s/%s_%d.weights", backup_directory, base, i);
            save_weights(net, buff);
        }
        if(i%100==0){
            char buff[256];
            sprintf(buff, "%s/%s.backup", backup_directory, base);
            save_weights(net, buff);
        }
        free_data(train);
    }
    char buff[256];
    sprintf(buff, "%s/%s_final.weights", backup_directory, base);
    save_weights(net, buff);
}

// Assembly:
pushq %rbp
pushq %r15
pushq %r14
pushq %r13
pushq %r12
pushq %rbx
subq $0x318, %rsp            # imm = 0x318
movl %edx, %ebp
movq %rsi, %r14
movq %rdi, %r15
xorl %edi, %edi
callq 0x32c0
movl %eax, %edi
callq 0x3220
movq %r15, %rdi
callq 0x41c8
movq %rax, 0xb0(%rsp)
movq %rax, %rdi
callq 0x30b0
movq %r15, %rdi
movq %r14, %rsi
movl %ebp, %edx
callq 0x22658
movq %rax, %r14
cvtss2sd 0x34(%rax), %xmm0
cvtss2sd 0x38(%rax), %xmm1
cvtss2sd 0x3c(%rax), %xmm2
leaq 0x194eb(%rip), %rdi     # 0x50ed0
movb $0x3, %al
callq 0x3160
movslq 0x4(%r14), %rax
movslq 0x1c(%r14), %rbx
imulq %rax, %rbx
movq 0x8(%r14), %rax
movq (%rax), %rax
xorl %edx, %edx
divq %rbx
movq %rax, %r15
leaq 0x1966e(%rip), %rdi     # 0x5107c
callq 0x1c8c8
movq %rax, %r12
movq %rax, %rdi
callq 0x72f5
movq %rax, %r13
leaq 0x168(%rsp), %rbp
movl $0xa8, %edx
movq %rbp, %rdi
xorl %esi, %esi
callq 0x31c0
movq 0x90(%r14), %rax
movq %rax, 0x28(%rbp)
movl $0x4, 0x58(%rbp)
movq %r13, 0x8(%rbp)
movl %ebx, 0x18(%rbp)
movl (%r12), %eax
movl %eax, 0x1c(%rbp)
leaq 0x128(%rsp), %rax
movq %rax, 0x80(%rbp)
movl $0xc, 0x98(%rbp)
pushq $0x15
popq %rcx
movq %rsp, %rdi
movq %rbp, %rsi
rep  movsq (%rsi), %es:(%rdi)
movss 0x147df(%rip), %xmm0    # 0x4c264
movaps %xmm0, 0xd0(%rsp)
callq 0x20b03
movq %rax, %r13
leaq 0x210(%rsp), %rbp
movq %r14, %rdi
callq 0x22693
movslq 0x54(%r14), %rcx
cmpq %rcx, %rax
jae 0x37cf0
incl %r15d
callq 0x3120
movq %rax, %r12
movq %r13, %rdi
xorl %esi, %esi
callq 0x3470
movups 0x128(%rsp), %xmm0
movups 0x138(%rsp), %xmm1
movups 0x148(%rsp), %xmm2
movups 0x158(%rsp), %xmm3
movaps %xmm0, 0xe0(%rsp)
movaps %xmm1, 0xf0(%rsp)
movaps %xmm2, 0x100(%rsp)
movaps %xmm3, 0x110(%rsp)
pushq $0x15
popq %rcx
movq %rsp, %rdi
leaq 0x168(%rsp), %rsi
rep  movsq (%rsi), %es:(%rdi)
callq 0x20b03
movq %rax, %r13
callq 0x3120
subq %r12, %rax
movq %rax, %rdi
callq 0x437c
cvtss2sd %xmm0, %xmm0
leaq 0x19403(%rip), %rdi     # 0x50f3e
movb $0x1, %al
callq 0x3160
callq 0x3120
movq %rax, %r12
movaps 0xe0(%rsp), %xmm0
movaps 0xf0(%rsp), %xmm1
movaps 0x100(%rsp), %xmm2
movaps 0x110(%rsp), %xmm3
movups %xmm3, 0x30(%rsp)
movups %xmm2, 0x20(%rsp)
movups %xmm1, 0x10(%rsp)
movups %xmm0, (%rsp)
movq %r14, %rdi
callq 0x22df8
movaps 0xd0(%rsp), %xmm3
movaps %xmm3, %xmm1
cmpltss 0x144df(%rip), %xmm1    # 0x4c078
movaps %xmm1, %xmm2
andnps %xmm3, %xmm2
andps %xmm0, %xmm1
orps %xmm2, %xmm1
cvtss2sd %xmm1, %xmm1
cvtss2sd %xmm0, %xmm0
movsd %xmm0, 0xc8(%rsp)
mulsd 0x17a12(%rip), %xmm0    # 0x4f5d0
mulsd 0x19302(%rip), %xmm1    # 0x50ec8
addsd %xmm0, %xmm1
cvtsd2ss %xmm1, %xmm0
movaps %xmm0, 0xd0(%rsp)
cvtss2sd %xmm0, %xmm0
movsd %xmm0, 0xc0(%rsp)
movq %r14, %rdi
callq 0x226ae
cvtss2sd %xmm0, %xmm0
movsd %xmm0, 0xb8(%rsp)
callq 0x3120
subq %r12, %rax
movq %rax, %rdi
callq 0x437c
cvtss2sd %xmm0, %xmm3
movl %r15d, %edx
imull %ebx, %edx
leaq 0x194c3(%rip), %rdi     # 0x510dc
movl %r15d, %esi
movsd 0xc8(%rsp), %xmm0
movsd 0xc0(%rsp), %xmm1
movsd 0xb8(%rsp), %xmm2
movb $0x4, %al
callq 0x3160
movl %r15d, %eax
cltd
movl $0x3e8, %ecx            # imm = 0x3E8
idivl %ecx
testl %edx, %edx
jne 0x37c7b
movq %rbp, %rdi
leaq 0x194b6(%rip), %rsi     # 0x5110d
leaq 0x19467(%rip), %rdx     # 0x510c5
movq 0xb0(%rsp), %rcx
movl %r15d, %r8d
xorl %eax, %eax
callq 0x3420
movq %r14, %rdi
movq %rbp, %rsi
callq 0x28aed
movl %r15d, %eax
cltd
pushq $0x64
popq %rcx
idivl %ecx
testl %edx, %edx
jne 0x37cb3
movq %rbp, %rdi
leaq 0x1948c(%rip), %rsi     # 0x5111e
leaq 0x1942c(%rip), %rdx     # 0x510c5
movq 0xb0(%rsp), %rcx
xorl %eax, %eax
callq 0x3420
movq %r14, %rdi
movq %rbp, %rsi
callq 0x28aed
movaps 0xe0(%rsp), %xmm0
movaps 0xf0(%rsp), %xmm1
movaps 0x100(%rsp), %xmm2
movaps 0x110(%rsp), %xmm3
movups %xmm3, 0x30(%rsp)
movups %xmm2, 0x20(%rsp)
movups %xmm1, 0x10(%rsp)
movups %xmm0, (%rsp)
callq 0x1e69d
jmp 0x37a9d
leaq 0x19434(%rip), %rsi     # 0x5112b
leaq 0x193c7(%rip), %rdx     # 0x510c5
leaq 0x210(%rsp), %r15
movq %r15, %rdi
movq 0xb0(%rsp), %rcx
xorl %eax, %eax
callq 0x3420
movq %r14, %rdi
movq %r15, %rsi
callq 0x28aed
addq $0x318, %rsp            # imm = 0x318
popq %rbx
popq %r12
popq %r13
popq %r14
popq %r15
popq %rbp
retq