Open bwmr opened 3 hours ago
I tested giving --arch unet_5
and --arch ressmall_18
as parameters, and it also gives an error:
Traceback (most recent call last): File "simsiam_main.py", line 170, in
main(opt) File "simsiam_main.py", line 118, in main log_dicttrain, , = trainer.train(epoch, train_loader) File "/home/Medalia/BWimmer/Scripts/cet_pick/cet_pick/trains/base_trainer.py", line 577, in train return self.run_epoch('train', epoch, data_loader) File "/home/Medalia/BWimmer/Scripts/cet_pick/cet_pick/trains/base_trainer.py", line 505, in run_epoch output, loss, loss_stats = model_with_loss(batch, epoch, phase) File "/home/Medalia/BWimmer/micromamba/envs/milopyp/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl return forward_call(*args, *kwargs) File "/home/Medalia/BWimmer/Scripts/cet_pick/cet_pick/trains/base_trainer.py", line 117, in forward outputs = self.model(batch['input'], batch['input_3d'], batch['input_aug'], batch['input_aug_3d']) File "/home/Medalia/BWimmer/micromamba/envs/milopyp/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl return forward_call(args, **kwargs) TypeError: forward() takes 2 positional arguments but 5 were given
Hi, I'm interesting in testing MiLoPYP. Currently, I'm trying to replicate the globular particle picking tutorial. However, when I run the first command as listed in the documentation
python simsiam_main.py simsiam2d3d --num_epochs 20 --exp_id test_sample --bbox 36 --dataset simsiam2d3d --arch simsiam2d3d_18 --lr 1e-3 --train_img_txt sample_train_explore_img.txt --batch_size 256 --val_intervals 20 --save_all --gauss 0.8 --dog 3,5
I receive the error message below. Training still starts, however. When checking the available architectures via
python simsiam_main.py smsiam2d3d --help
, the available architectures areressmall_18 | unet_4 | unet_5
.Do I need additional model files for the
simsiam2d3d_18
architecture, or is the quickstart protocol or help outdated?Thanks and best regards, Benedikt