Skip to content

Update pytorch version in USAGE.md#21

Open
qwzybug wants to merge 1 commit into
NVlabs:masterfrom
qwzybug:patch-1
Open

Update pytorch version in USAGE.md#21
qwzybug wants to merge 1 commit into
NVlabs:masterfrom
qwzybug:patch-1

Conversation

@qwzybug

@qwzybug qwzybug commented May 29, 2018

Copy link
Copy Markdown

Running the examples with pytorch 0.3 causes an error:

(img2img) paperspace@psgg6az8h:~/img2img/MUNIT$ python test.py --config configs/edges2handbags_folder.yaml --input inputs/edge.jpg --output_folder outputs --checkpoint models/edges2handbags.pt --a2b 1
Traceback (most recent call last):
  File "test.py", line 48, in <module>
    trainer = MUNIT_Trainer(config)
  File "/home/paperspace/img2img/MUNIT/trainer.py", line 17, in __init__
    self.gen_a = AdaINGen(hyperparameters['input_dim_a'], hyperparameters['gen'])  # auto-encoder for domain a
  File "/home/paperspace/img2img/MUNIT/networks.py", line 105, in __init__
    self.enc_content = ContentEncoder(n_downsample, n_res, input_dim, dim, 'in', activ, pad_type=pad_type)
  File "/home/paperspace/img2img/MUNIT/networks.py", line 210, in __init__
    self.model += [Conv2dBlock(input_dim, dim, 7, 1, 3, norm=norm, activation=activ, pad_type=pad_type)]
  File "/home/paperspace/img2img/MUNIT/networks.py", line 308, in __init__
    self.norm = nn.InstanceNorm2d(norm_dim, track_running_stats=True)
TypeError: __init__() got an unexpected keyword argument 'track_running_stats'

track_running_stats isn't defined until pytorch > 0.3.1.
pytorch:nn.modules.instancenorm@0.3.1
pytorch:nn.modules.instancenorm@0.4.0

Running the examples with pytorch 0.3 causes an error:

```
(img2img) paperspace@psgg6az8h:~/img2img/MUNIT$ python test.py --config configs/edges2handbags_folder.yaml --input inputs/edge.jpg --output_folder outputs --checkpoint models/edges2handbags.pt --a2b 1
Traceback (most recent call last):
  File "test.py", line 48, in <module>
    trainer = MUNIT_Trainer(config)
  File "/home/paperspace/img2img/MUNIT/trainer.py", line 17, in __init__
    self.gen_a = AdaINGen(hyperparameters['input_dim_a'], hyperparameters['gen'])  # auto-encoder for domain a
  File "/home/paperspace/img2img/MUNIT/networks.py", line 105, in __init__
    self.enc_content = ContentEncoder(n_downsample, n_res, input_dim, dim, 'in', activ, pad_type=pad_type)
  File "/home/paperspace/img2img/MUNIT/networks.py", line 210, in __init__
    self.model += [Conv2dBlock(input_dim, dim, 7, 1, 3, norm=norm, activation=activ, pad_type=pad_type)]
  File "/home/paperspace/img2img/MUNIT/networks.py", line 308, in __init__
    self.norm = nn.InstanceNorm2d(norm_dim, track_running_stats=True)
TypeError: __init__() got an unexpected keyword argument 'track_running_stats'
```

`track_running_stats` isn't defined until pytorch > 0.3.1.
[pytorch:nn.modules.instancenorm@0.3.1](https://github.com/pytorch/pytorch/blob/v0.3.1/torch/nn/modules/instancenorm.py)
[pytorch:nn.modules.instancenorm@0.4.0](https://github.com/pytorch/pytorch/blob/v0.4.0/torch/nn/modules/instancenorm.py)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

1 participant