| 57 | top[0].data[()] = self.phase |
| 58 | |
| 59 | def python_net_file(): |
| 60 | with tempfile.NamedTemporaryFile(mode='w+', delete=False) as f: |
| 61 | f.write("""name: 'pythonnet' force_backward: true |
| 62 | input: 'data' input_shape { dim: 10 dim: 9 dim: 8 } |
| 63 | layer { type: 'Python' name: 'one' bottom: 'data' top: 'one' |
| 64 | python_param { module: 'test_python_layer' layer: 'SimpleLayer' } } |
| 65 | layer { type: 'Python' name: 'two' bottom: 'one' top: 'two' |
| 66 | python_param { module: 'test_python_layer' layer: 'SimpleLayer' } } |
| 67 | layer { type: 'Python' name: 'three' bottom: 'two' top: 'three' |
| 68 | python_param { module: 'test_python_layer' layer: 'SimpleLayer' } }""") |
| 69 | return f.name |
| 70 | |
| 71 | |
| 72 | def exception_net_file(): |