David310/Detect_AI-generated_Image
4
1U
2 ��^f� � @ s d dl mZ G dd� de�ZdS )� )� BaseOptionsc @ s e Zd Zdd� ZdS )� TrainOptionsc C s� t �| |�}|jdtdd� |jdddd� |jdtd d3d � |jd dd
d� |jdtddd � |jdtddd � |jdtddd � |jdtddd � |jdtddd � |jd td dd � |jd td d d � |jd!td"d#d � |jd$td%d&d � d'| _|S )(Nz--earlystop_epoch� )�type�defaultz4--data_aug�5store_truezTif specified, perform additional data augmentation (photometric, blurring, jpegging))�action�helpz--optim�adamzoptim to use [sgd, adam])r r r z --new_optimz0new optimizer instead of loading the optim statez --loss_freqi� z(frequency of showing loss on tensorboardz--save_epoch_freqr z4frequency of saving checkpoints at the end of epochsz
--epoch_countzcthe starting epoch count, we save the model by <epoch_count>, <epoch_count>+<save_latest_freq>, ...z --last_epoch�����z0starting epoch count for scheduler intializationz
--train_split�trainztrain, val, test, etcz --val_split�valz--niter�d z
total epochesz--beta1g�������?zmomentum term of adamz--lrg-C ��6?z initial learning rate for adamT)r �6initialize� add_argument�int�str�float�isTrain)�self�parser� r �=/root/autodl-tmp/UniversalFakeDetect/options/train_options.pyr s zTrainOptions.initializeN)�__name__�7__module__� __qualname__r r r r r r s r N)Z base_optionsr r r r r r �<module> s