-
Notifications
You must be signed in to change notification settings - Fork 1
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
issue with the code #3
Comments
It seems losslmse never be culculated |
Hi I am not sure about the issue. |
The unetacresnet.bash script works, but there are still some problems, such as the loss does not drop, and there is no way to run the unet model successfully.
|
The parameters have been set to work well on my dataset. Try to reduce the learning rate to 10^-5, change batch size. You can also add 1 or 2 skip connections. If you need latent space features encode to more information, increase the latent_loss parameter. Maybe you need to modify the loss, or change the number of layers. |
Hi I'm running training process using unet.bash, but during training a bug was encountered.
the unet.bash is like:
CUDA_VISIBLE_DEVICES=0 python3 main.py --mode train --train_file lists/training.txt \ --valid_file lists/validation.txt \ --test_file lists/testing.txt --batch_size 16 --sample_length 1 \ --total_length 1 --number_of_crops 1 --buffer_size 100 --exp_name uvae-test1-0810 --learning_rate 0.0008 \ --checkpoint_dir ./data/checkpoints/ --model UNet --datatype outdoor --num_epochs 300 \ --num_class 10 --block_size 1 --probability 0
and the error is like:
AttributeError: 'NoneType' object has no attribute 'log_scalar'
and I trace this error to:
` self.logger.log_scalar('l2 loss', self.losslmse
could you help me with it?
thanks alot
The text was updated successfully, but these errors were encountered: