search

michaal94 / torch_DCEC

Pytorch Deep Clustering with Convolutional Autoencoders implementation
MIT License
100 stars 30 forks source link

Pretrain mode Parameters #2

Closed fina1e closed 5 years ago

fina1e commented 5 years ago

At main, the Parameter of pretrain should be instead:

    elif args.mode == 'pretrain':
        model = training_functions.pretraining(model, dataloader, criteria, optimizers, schedulers, epochs, params)

either indexed at passing or at pretrain:

model = training_functions.pretraining(model, dataloader1,dataloader2, criteria[0], optimizers[1], schedulers[1], epochs, params)
michaal94 commented 5 years ago

Changed in last commit. Thanks